{
 "cells": [
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "# torch.nn.functional"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "## Convolution 函数\n",
    "### torch.nn.functional.conv1d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)\n",
    "\n",
    "对几个输入平面组成的输入信号应用1D（一维）卷积。 有关详细信息和输出形状，请参见 Conv1d 。\n",
    "\n",
    "参数： \n",
    "- input – 输入张量的形状 (minibatch x in_channels x iW) \n",
    "- weight – 过滤器的形状 (out_channels, in_channels, kW) \n",
    "- bias – 可选偏置的形状 (out_channels) \n",
    "- stride – 卷积核的步长，默认为1"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 1,
   "metadata": {},
   "outputs": [],
   "source": [
    "import torch\n",
    "from torch import autograd\n",
    "import torch.nn.functional as F"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 2,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "tensor([[[ 4.0910e+00,  1.2746e+00,  1.3093e+01,  ..., -5.6192e+00,\n",
       "           1.5224e+00,  5.8159e+00],\n",
       "         [ 1.2228e+01,  1.1855e+01,  4.6436e+00,  ...,  3.2987e+00,\n",
       "           5.8506e+00,  4.1566e+00],\n",
       "         [-1.7404e+00, -5.3658e+00,  2.6954e+00,  ..., -3.3302e-01,\n",
       "           6.2116e+00,  4.6132e+00],\n",
       "         ...,\n",
       "         [-8.7469e+00,  8.6097e+00, -4.1213e+00,  ...,  9.0458e-02,\n",
       "          -1.2159e+01, -7.5299e+00],\n",
       "         [ 5.8603e+00, -5.6252e+00,  3.2814e+00,  ..., -5.6246e-01,\n",
       "           2.9551e+00,  1.8501e+01],\n",
       "         [-8.9731e+00, -7.8859e-01,  1.1931e+01,  ...,  2.7926e+00,\n",
       "           8.3333e-01, -1.4514e+01]],\n",
       "\n",
       "        [[ 4.8804e+00,  6.8274e+00,  1.3089e+00,  ...,  7.3061e+00,\n",
       "          -1.9042e+00, -2.7113e+00],\n",
       "         [ 5.0816e+00,  2.1482e+00, -1.1752e+01,  ..., -1.7421e+00,\n",
       "           1.0823e+00,  1.0024e+01],\n",
       "         [-1.5022e+01, -6.0568e-01,  3.2907e+00,  ..., -7.0923e-01,\n",
       "          -2.2644e+00,  2.5486e+00],\n",
       "         ...,\n",
       "         [-5.0079e-01, -1.1079e+01,  1.3402e+01,  ...,  9.4584e-01,\n",
       "           6.2116e+00,  2.1180e-01],\n",
       "         [-3.6502e+00,  7.6502e-01, -1.4066e+01,  ..., -4.5368e+00,\n",
       "          -1.0073e+01,  3.2376e+00],\n",
       "         [-6.4932e+00,  4.6820e+00,  3.4953e+00,  ...,  7.5259e+00,\n",
       "           9.5683e+00,  1.3552e+01]],\n",
       "\n",
       "        [[-3.0305e+00, -7.4078e+00,  8.5769e+00,  ...,  6.0557e-01,\n",
       "          -2.1675e+00, -1.3704e-01],\n",
       "         [-3.7021e+00,  5.2522e+00, -1.8131e-01,  ..., -7.0846e+00,\n",
       "           5.4569e+00, -3.7227e+00],\n",
       "         [-5.4512e+00,  4.6566e-01,  6.6060e-01,  ...,  4.3908e+00,\n",
       "          -4.4050e-01,  3.5189e+00],\n",
       "         ...,\n",
       "         [ 4.2786e+00, -5.5746e+00,  2.2203e+00,  ..., -7.2806e+00,\n",
       "           4.1890e+00,  4.6285e-01],\n",
       "         [ 3.1481e+00, -8.3490e+00, -6.3298e+00,  ..., -6.3448e-01,\n",
       "          -4.4952e+00, -1.0193e+01],\n",
       "         [ 7.9670e+00,  4.2885e+00, -7.9932e-01,  ...,  6.3337e+00,\n",
       "          -6.9233e-01,  1.1588e+01]],\n",
       "\n",
       "        ...,\n",
       "\n",
       "        [[ 8.4968e+00,  2.4855e-01, -1.3254e-01,  ...,  3.7633e+00,\n",
       "          -6.4758e-01, -2.3723e-01],\n",
       "         [-7.7712e+00,  3.5705e+00, -8.5080e-01,  ...,  5.0773e-01,\n",
       "          -4.4580e+00,  1.1213e+01],\n",
       "         [ 1.3327e+01, -3.4302e+00,  9.1954e+00,  ..., -5.1101e+00,\n",
       "           1.7082e+00,  1.2456e+01],\n",
       "         ...,\n",
       "         [-9.0585e+00,  4.6722e+00,  7.0620e+00,  ..., -1.4885e+00,\n",
       "           6.4596e+00,  7.7810e+00],\n",
       "         [-1.0348e+00, -2.3381e+00, -1.6548e+01,  ..., -7.0781e+00,\n",
       "          -5.0067e+00,  2.8779e-01],\n",
       "         [-1.4200e+00,  1.6257e+01, -3.3983e-01,  ...,  1.9043e+00,\n",
       "          -3.5639e+00,  3.6113e+00]],\n",
       "\n",
       "        [[ 2.6513e+00,  8.8154e+00, -2.5958e+00,  ..., -4.3953e-01,\n",
       "          -1.3690e+00,  2.9913e+00],\n",
       "         [ 1.0256e+01, -8.8433e+00, -7.0052e+00,  ...,  1.3082e+01,\n",
       "           1.3005e+01, -1.9345e-02],\n",
       "         [ 1.3775e+01, -1.0761e+00, -9.4317e+00,  ..., -4.6827e+00,\n",
       "          -1.2804e+01,  8.2587e+00],\n",
       "         ...,\n",
       "         [-1.3406e+00,  3.3186e+00, -2.8816e-01,  ..., -6.8477e+00,\n",
       "          -5.1220e+00,  3.7377e-01],\n",
       "         [ 1.9998e+00, -5.1414e+00, -3.4086e+00,  ...,  1.9606e+01,\n",
       "           1.1927e+01, -1.2944e+01],\n",
       "         [-1.4355e+01, -3.4957e+00, -6.3899e-01,  ..., -4.1978e-01,\n",
       "           5.1508e+00, -3.3365e+00]],\n",
       "\n",
       "        [[ 2.4047e+00,  3.6760e+00,  5.9617e+00,  ...,  4.7167e+00,\n",
       "           1.3831e+00, -8.8700e+00],\n",
       "         [ 6.8102e+00, -3.6528e-01, -8.3459e+00,  ...,  3.1406e+00,\n",
       "          -4.5077e+00,  4.5912e+00],\n",
       "         [-1.2930e+00,  8.1822e+00, -4.6832e+00,  ..., -3.1643e+00,\n",
       "          -5.9580e-01,  1.9888e+00],\n",
       "         ...,\n",
       "         [-6.1210e+00, -1.5999e+00,  1.1068e+01,  ...,  3.0141e+00,\n",
       "           3.1293e+00,  3.0049e+00],\n",
       "         [ 4.7406e+00,  1.1598e+00, -6.5645e+00,  ...,  4.3008e+00,\n",
       "           6.2542e+00,  2.0070e+00],\n",
       "         [-8.5456e+00,  1.1562e+01,  3.3210e+00,  ..., -3.6710e+00,\n",
       "           1.2002e+00, -1.0852e-01]]])"
      ]
     },
     "execution_count": 2,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "filters = autograd.Variable(torch.randn(33, 16, 3))\n",
    "inputs = autograd.Variable(torch.randn(20, 16, 50))\n",
    "F.conv1d(inputs, filters)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### torch.nn.functional.conv2d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)\n",
    "\n",
    "对几个输入平面组成的输入信号应用2D卷积。 有关详细信息和输出形状，请参见 Conv2d 。\n",
    "\n",
    "参数：\n",
    "- input – 输入张量 (minibatch x in_channels x iH x iW) \n",
    "- weight – 过滤器张量 (out_channels, in_channels/groups, kH, kW) \n",
    "- bias – 可选偏置张量 (out_channels) \n",
    "- stride – 卷积核的步长，可以是单个数字或一个元组 (sh x sw)。默认为1 \n",
    "- padding – 输入上隐含零填充。 可以是单个数字或元组。 默认值：0 \n",
    "- groups – 将输入分成组，in_channels应该被组数除尽"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "data": {
      "text/plain": [
       "tensor([[[[-5.4119e-02,  3.3753e+00,  6.7146e-01, -8.5523e+00,  6.5412e+00],\n",
       "          [ 3.7080e-01, -3.1662e+00, -5.3672e-01,  7.3597e+00,  8.3691e+00],\n",
       "          [-1.4331e+00, -5.8819e-01,  1.0410e+01,  5.3745e+00, -2.4462e+00],\n",
       "          [-1.2571e+01,  1.3455e-02, -4.1606e+00, -4.6117e+00,  4.9175e+00],\n",
       "          [-8.2907e+00, -2.0928e+00, -6.8445e+00,  5.8670e+00,  2.2476e+00]],\n",
       "\n",
       "         [[-1.1513e+00,  5.1455e+00,  1.5351e+00, -1.4796e+00, -2.0224e+00],\n",
       "          [-5.6302e+00, -5.5882e+00,  8.6554e+00, -3.3394e+00,  2.3090e+00],\n",
       "          [ 1.8197e+00,  1.2870e+00, -4.3271e-01, -1.8481e+00,  1.1259e+00],\n",
       "          [-9.5713e+00,  3.3243e+00, -1.4797e+00,  2.5831e+00,  3.3286e+00],\n",
       "          [ 1.7168e+00, -7.4784e+00,  3.5701e+00, -1.9641e+00, -3.3731e+00]],\n",
       "\n",
       "         [[-5.2279e-01, -4.6187e+00, -8.7876e-01,  1.6635e+00, -2.1674e+00],\n",
       "          [ 2.9052e+00,  1.5868e+00,  6.4764e+00,  8.5155e+00, -1.5919e+01],\n",
       "          [ 3.1077e+00,  1.1326e+01, -2.9387e+00, -1.0763e+01, -3.4723e+00],\n",
       "          [ 5.0171e+00, -5.9792e+00,  1.6614e+00,  1.4831e+01,  4.4058e+00],\n",
       "          [ 9.6134e-01, -1.0315e+00,  1.9431e+00,  4.5330e+00, -5.1682e+00]],\n",
       "\n",
       "         [[ 4.1710e+00, -5.7725e+00, -1.7835e+00,  1.4133e+01,  1.8198e+00],\n",
       "          [-2.9769e+00,  1.9638e+00,  1.0241e+01,  7.2867e+00,  6.0526e-02],\n",
       "          [ 2.0425e+00,  3.7125e+00,  5.9494e+00,  4.7634e+00,  9.5722e-01],\n",
       "          [-2.7725e+00,  6.9019e+00,  4.4944e+00,  4.0105e-01,  1.9717e+00],\n",
       "          [-4.3724e+00, -9.4162e+00, -5.0506e-01, -3.1622e-03,  5.9092e-02]],\n",
       "\n",
       "         [[-4.9458e+00,  4.2540e+00, -7.4914e+00,  6.2418e+00,  4.8843e+00],\n",
       "          [ 2.7867e+00, -9.1320e+00,  1.1018e+01,  2.0888e+00, -6.8002e+00],\n",
       "          [-7.8459e+00, -2.8481e+00,  5.7970e+00, -3.3005e+00, -3.0223e+00],\n",
       "          [-8.9508e+00, -5.5132e+00, -4.7996e+00,  3.8179e+00,  5.6929e+00],\n",
       "          [ 7.5595e+00, -3.8432e+00, -5.0050e+00, -8.6614e+00,  6.1119e-01]],\n",
       "\n",
       "         [[ 5.5191e-01,  4.2519e-02,  3.7280e+00, -5.9032e+00,  4.4994e-01],\n",
       "          [ 3.0031e+00, -3.8100e+00,  4.9460e+00,  4.7643e+00, -5.7894e+00],\n",
       "          [ 1.6832e+00,  3.6856e-01, -6.6175e+00, -3.2084e+00, -4.0319e-01],\n",
       "          [-4.7335e+00,  3.9306e+00, -9.4926e+00, -5.8364e+00, -5.8681e+00],\n",
       "          [ 3.4080e+00, -1.7597e-01,  4.4352e+00,  7.5870e+00, -5.9266e+00]],\n",
       "\n",
       "         [[-1.0144e+00, -2.8548e+00,  1.3715e+00, -3.9042e+00,  2.8022e-01],\n",
       "          [-4.3729e+00,  8.8594e+00,  1.2637e+01,  3.7179e+00,  8.9754e+00],\n",
       "          [ 9.4860e-01, -5.9226e+00,  1.3922e-01, -5.4075e+00, -6.5756e+00],\n",
       "          [ 1.1597e+01, -7.4384e+00, -6.7793e+00, -4.4545e-02,  8.9132e+00],\n",
       "          [-5.4914e+00,  8.9997e+00,  1.2083e+01, -2.9189e+00,  1.4173e+00]],\n",
       "\n",
       "         [[-3.6300e+00,  2.8323e+00, -3.0684e+00,  5.8751e+00, -1.2132e+00],\n",
       "          [ 1.3078e+00, -8.1079e+00,  8.4842e+00,  3.1408e+00,  2.6241e+00],\n",
       "          [ 2.2583e+00,  8.5299e+00,  2.1670e+00, -3.7853e+00, -7.1620e+00],\n",
       "          [ 6.6417e+00,  6.5304e+00,  1.5158e+00, -1.1240e+01, -1.7694e+00],\n",
       "          [-1.9263e+00, -3.6768e+00,  1.5842e+00,  7.0705e+00,  3.6415e+00]]]])"
      ]
     },
     "execution_count": 3,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "# With square kernels and equal stride\n",
    "filters = autograd.Variable(torch.randn(8,4,3,3))\n",
    "inputs = autograd.Variable(torch.randn(1,4,5,5))\n",
    "F.conv2d(inputs, filters, padding=1)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### torch.nn.functional.conv3d(input, weight, bias=None, stride=1, padding=0, dilation=1, groups=1)\n",
    "\n",
    "对几个输入平面组成的输入信号应用3D卷积。 有关详细信息和输出形状，请参见 Conv3d 。\n",
    "\n",
    "参数： \n",
    "- input – 输入张量的形状 (minibatch x in_channels x iT x iH x iW) \n",
    "- weight – 过滤器张量 的形状 (out_channels, in_channels, kT, kH, kW) \n",
    "- bias – 可选偏置张量的形状 (out_channels) \n",
    "- stride – 卷积核的步长，可以是单个数字或一个元组 (sh x sw)。默认为1 \n",
    "- padding – 输入上隐 含零填充。可以是单个数字或元组。 默认值：0"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [
    {
     "data": {
      "text/plain": [
       "tensor([[[[[ 4.1161e+01, -1.0434e+01, -1.2643e+01,  ..., -1.6542e+01,\n",
       "            -1.4606e+01,  1.4156e+01],\n",
       "           [ 4.9931e+00, -2.2831e+01, -1.2825e+01,  ..., -6.7927e+00,\n",
       "             1.1988e+01,  1.5875e+01],\n",
       "           [-3.2071e+01,  5.2264e+00, -1.1677e+01,  ...,  1.0813e+01,\n",
       "             2.7621e+00,  1.1468e+01],\n",
       "           ...,\n",
       "           [-4.5656e+01, -5.4573e+00,  1.5794e+01,  ..., -1.7488e+01,\n",
       "             4.1186e+01, -3.8321e+01],\n",
       "           [ 2.6206e+00,  7.7757e+00, -2.3285e+01,  ..., -5.1505e+00,\n",
       "             3.0232e+00, -1.4973e+00],\n",
       "           [ 1.6142e+01, -1.0456e+01, -2.8018e+01,  ...,  3.8569e+01,\n",
       "            -5.4558e+00,  1.4217e+01]],\n",
       "\n",
       "          [[ 3.4312e+01,  4.3441e+00, -9.3609e+00,  ...,  9.3957e+00,\n",
       "             1.5157e+00,  4.4296e+01],\n",
       "           [ 1.4458e+01,  3.1759e+00,  1.5342e+01,  ..., -1.6613e+01,\n",
       "            -3.1984e+01,  9.5534e+00],\n",
       "           [-2.5717e+01, -2.6193e+01,  6.0278e+00,  ..., -2.4641e+01,\n",
       "             4.0979e+00,  1.0392e+00],\n",
       "           ...,\n",
       "           [ 2.6219e+01, -6.1337e+01,  1.2964e+01,  ..., -1.7186e+01,\n",
       "            -1.0775e+01, -1.0307e+00],\n",
       "           [-8.6409e+00, -3.8810e+01, -3.3777e+01,  ...,  3.0419e+01,\n",
       "             1.1130e+01,  1.3278e+01],\n",
       "           [-4.2323e+01,  4.5708e+00,  2.3941e-01,  ...,  1.1620e+01,\n",
       "            -6.8273e+00, -2.4681e+01]],\n",
       "\n",
       "          [[-1.5229e+00,  4.8765e+00,  4.5764e+01,  ..., -1.9986e+01,\n",
       "            -3.7781e+01, -1.2039e+01],\n",
       "           [-6.4055e+00, -2.4686e+00, -3.7642e+00,  ...,  3.5719e+00,\n",
       "             4.1146e+00,  4.7017e+01],\n",
       "           [ 1.5050e+01,  2.7754e+01, -8.5313e+00,  ..., -6.2413e+00,\n",
       "             6.0562e+00, -2.5612e+00],\n",
       "           ...,\n",
       "           [ 1.3727e+01,  1.8567e+00, -1.4776e+01,  ...,  3.9007e+01,\n",
       "             1.2247e+01, -9.8180e+00],\n",
       "           [ 6.0396e+00, -2.5214e+00,  1.3784e+01,  ..., -1.3717e+01,\n",
       "            -2.4973e+01,  2.4472e+01],\n",
       "           [ 3.1267e+01, -1.7024e+01, -2.0207e+01,  ..., -3.9725e+00,\n",
       "            -2.1282e+01,  1.1155e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-8.4898e+00, -9.8195e+00,  7.8822e+00,  ...,  1.1315e+01,\n",
       "             2.5564e+01,  3.0882e+01],\n",
       "           [ 8.7242e+00, -2.0592e+01, -3.4539e+01,  ...,  1.7535e+01,\n",
       "            -9.3941e+00, -8.9942e+00],\n",
       "           [-3.7704e+01,  1.7678e+01, -3.0470e+00,  ..., -3.8909e+01,\n",
       "             1.7218e+01, -2.6866e+01],\n",
       "           ...,\n",
       "           [ 7.2002e-01,  7.6702e+00,  1.8439e+01,  ...,  1.2651e+01,\n",
       "             3.7863e+01,  1.6482e+01],\n",
       "           [-2.5462e+01,  1.0246e+01,  1.0295e+01,  ..., -1.0934e+01,\n",
       "            -2.9798e+01,  1.8612e+01],\n",
       "           [-2.7338e+01, -1.1191e+01, -8.1940e+00,  ..., -4.7747e+01,\n",
       "            -1.1372e+01, -7.7026e+00]],\n",
       "\n",
       "          [[-1.5492e+01, -2.1254e+01,  2.4005e+01,  ..., -1.6579e+01,\n",
       "            -2.3681e+01,  1.8621e+01],\n",
       "           [-1.0431e+01, -2.1647e+01,  6.5477e+00,  ..., -1.9737e+01,\n",
       "             1.1265e+01,  7.1010e+00],\n",
       "           [-2.2828e-01,  2.1567e+00, -3.7608e+01,  ...,  1.2963e+01,\n",
       "            -1.0587e+01, -1.1286e+01],\n",
       "           ...,\n",
       "           [ 1.0737e+01,  5.7579e+00, -5.5726e+01,  ..., -3.8673e+01,\n",
       "            -1.9674e+01,  1.3634e+01],\n",
       "           [ 2.4038e+01, -1.7011e+01,  1.4721e+01,  ...,  1.5045e+01,\n",
       "            -3.5395e+01, -1.1894e+01],\n",
       "           [-4.4851e+00, -1.1969e+01, -8.7070e+00,  ..., -1.4483e+01,\n",
       "             1.4344e+01,  8.0755e+00]],\n",
       "\n",
       "          [[-1.3204e+00,  1.7344e+01,  1.8997e+01,  ...,  4.1615e+00,\n",
       "             5.5360e+00,  4.1543e+01],\n",
       "           [-3.2223e+01, -1.7374e+01,  7.6981e+00,  ..., -2.5270e+00,\n",
       "            -2.4755e+01,  1.4487e+01],\n",
       "           [-1.0239e+01,  2.8201e+01, -7.2031e+00,  ..., -5.2765e+01,\n",
       "            -1.0891e+00, -1.5112e+01],\n",
       "           ...,\n",
       "           [ 1.4034e+00, -1.2775e+01, -3.8388e+00,  ..., -4.5381e+00,\n",
       "             1.5530e+01,  2.0060e+00],\n",
       "           [ 5.4404e+01,  1.7566e+01,  7.6453e+00,  ..., -5.5323e+00,\n",
       "            -1.5397e+01, -9.8714e-01],\n",
       "           [ 1.3065e+00,  3.0539e+00, -5.5498e+00,  ...,  1.5198e+01,\n",
       "            -1.6166e+01,  1.3774e-01]]],\n",
       "\n",
       "\n",
       "         [[[ 1.2917e+01,  2.4754e+01,  2.8178e+00,  ...,  5.9275e+00,\n",
       "            -3.2149e+01, -1.4500e+01],\n",
       "           [-3.2088e+01,  5.3252e+00,  8.5143e+00,  ..., -7.8019e+00,\n",
       "             2.7203e+01,  1.6041e+01],\n",
       "           [ 8.9720e+00,  1.3531e+01,  8.5434e-01,  ..., -2.5376e+00,\n",
       "             2.1945e+01,  1.9731e+00],\n",
       "           ...,\n",
       "           [ 6.9776e+01, -4.9017e+00,  1.7082e+01,  ..., -1.3184e+01,\n",
       "             9.0950e+00,  1.8334e+01],\n",
       "           [-7.8452e+00, -4.6701e+00, -1.8183e+01,  ..., -5.3052e+01,\n",
       "             1.0580e+01, -1.9875e+01],\n",
       "           [-4.0720e+01, -4.6870e+01, -1.0006e+01,  ..., -2.3858e+01,\n",
       "            -1.5829e+01,  1.8449e+01]],\n",
       "\n",
       "          [[-1.0874e+01,  8.9387e+00,  3.2424e+00,  ...,  1.3737e+01,\n",
       "            -6.4593e+00, -2.7776e+01],\n",
       "           [ 8.4388e+00, -5.9463e+00,  8.1770e+00,  ..., -1.5150e+00,\n",
       "             3.8084e+01,  3.3436e+01],\n",
       "           [ 3.6543e+01,  2.3277e+01,  2.5729e+01,  ...,  1.4066e+01,\n",
       "            -1.7616e+00, -1.1425e+01],\n",
       "           ...,\n",
       "           [-2.2089e+01, -7.1791e+00,  4.5832e+00,  ..., -3.9434e+01,\n",
       "             3.6255e+01,  3.2083e+01],\n",
       "           [-1.3927e+01, -7.3518e+01, -1.0871e+01,  ...,  2.4676e+01,\n",
       "            -2.1304e+01,  2.2529e+00],\n",
       "           [ 3.7787e+00,  5.8200e+01, -2.7337e+00,  ..., -1.9304e+01,\n",
       "             5.7852e+00,  1.2134e+01]],\n",
       "\n",
       "          [[-7.5573e-02,  4.5005e+01,  8.4836e+00,  ...,  4.4138e+01,\n",
       "             3.0370e+01, -3.6062e+01],\n",
       "           [ 1.1854e+01, -1.8541e+01, -3.3361e+01,  ..., -7.4061e+00,\n",
       "            -3.4450e-01,  2.1944e+01],\n",
       "           [ 4.0727e+01, -3.5337e+01,  5.5726e+00,  ..., -9.5943e+00,\n",
       "             1.1904e+01, -3.9176e+00],\n",
       "           ...,\n",
       "           [-2.4905e+01, -1.5222e+01, -2.7657e+01,  ...,  1.4130e+01,\n",
       "             1.9919e-01, -2.7833e+01],\n",
       "           [-3.3360e+00,  6.6086e+00,  3.3307e+01,  ...,  1.8119e+01,\n",
       "             2.6690e+01,  5.6655e+00],\n",
       "           [-4.3616e+00, -4.6077e+00, -1.4753e+01,  ...,  1.5513e+01,\n",
       "             1.7907e+01, -1.2354e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-1.3746e+01,  3.7963e+00, -8.8374e+00,  ..., -3.1700e+00,\n",
       "             3.8201e+01,  2.7300e+00],\n",
       "           [-1.7183e+01, -4.0690e+01,  1.0890e+01,  ..., -3.0628e+01,\n",
       "             7.6687e+00, -1.3229e+01],\n",
       "           [ 2.4653e+01, -1.2038e+01, -2.0970e+01,  ...,  1.3635e+00,\n",
       "            -2.3713e+00, -1.0066e+01],\n",
       "           ...,\n",
       "           [ 7.2433e+00,  6.1918e+00,  1.2330e+01,  ...,  2.3438e+01,\n",
       "             3.6397e+00,  1.3131e+01],\n",
       "           [-1.4952e+01,  1.1784e+01, -3.3063e+01,  ..., -1.3500e+01,\n",
       "            -4.7556e+00,  2.6471e+00],\n",
       "           [ 1.6735e+01, -1.7531e+01,  8.7266e+00,  ..., -8.6433e+00,\n",
       "            -5.0567e+00,  7.0098e+00]],\n",
       "\n",
       "          [[-3.6823e+00, -2.6595e-02, -1.2365e+01,  ..., -1.0620e+01,\n",
       "             3.5646e+01, -9.6847e-01],\n",
       "           [ 2.6468e+01, -2.5323e+01,  1.3873e+01,  ...,  2.4982e+01,\n",
       "            -3.6796e+01,  2.4053e+01],\n",
       "           [ 4.2163e+01, -4.8428e+01, -6.1844e+00,  ..., -2.1573e+01,\n",
       "             9.3528e+00,  4.5555e+00],\n",
       "           ...,\n",
       "           [ 1.4377e+01, -1.9379e+01, -8.4064e+00,  ..., -4.6536e+01,\n",
       "            -5.2511e+00, -2.7787e+01],\n",
       "           [-2.0242e+01, -1.4933e+00,  2.0702e+01,  ...,  4.7282e+00,\n",
       "            -1.6682e+01,  6.4235e-02],\n",
       "           [-1.1359e+01, -4.5459e+00, -3.1970e+00,  ..., -6.9762e+00,\n",
       "            -1.6634e+00, -1.4943e+01]],\n",
       "\n",
       "          [[ 3.9342e+01, -1.1571e+01,  1.0667e+01,  ...,  4.7793e+00,\n",
       "             2.5091e+01,  2.3763e+01],\n",
       "           [-1.6364e+01, -1.0466e+01,  1.0773e+01,  ..., -8.5458e+00,\n",
       "            -1.6842e+01,  1.9556e+01],\n",
       "           [ 3.0544e+01,  2.7821e+00, -6.8297e+00,  ..., -1.3973e+01,\n",
       "             2.4785e+01,  3.0124e+01],\n",
       "           ...,\n",
       "           [-4.8748e+00, -1.0197e+01, -1.3098e+00,  ..., -9.1232e+00,\n",
       "             9.5013e+00,  1.7241e+01],\n",
       "           [-7.9008e+00,  1.5374e+01,  4.2089e+01,  ...,  9.0136e+00,\n",
       "            -1.7185e+01, -3.9619e+01],\n",
       "           [ 4.1547e+01,  2.1277e+01,  1.5898e+01,  ..., -7.2862e-01,\n",
       "             1.1439e+01,  1.1051e+01]]],\n",
       "\n",
       "\n",
       "         [[[ 6.4654e+00,  5.1011e+01,  9.8228e+00,  ...,  2.5362e+01,\n",
       "            -7.9973e+00,  3.4534e+01],\n",
       "           [ 1.8395e+01, -7.5412e+00,  6.0007e+01,  ..., -2.4824e+01,\n",
       "            -1.5067e+01, -6.7365e+00],\n",
       "           [-1.6326e+01, -2.4841e+01,  8.8982e+00,  ..., -1.1894e+01,\n",
       "             1.4650e+01,  2.0585e+01],\n",
       "           ...,\n",
       "           [-5.2208e+01, -9.1541e+00,  3.0995e+01,  ...,  8.8530e+00,\n",
       "            -5.1769e+01,  1.2627e+01],\n",
       "           [-1.4793e+01, -4.1322e+00,  2.9939e+00,  ...,  1.5361e+01,\n",
       "             1.6166e+01,  1.5379e+01],\n",
       "           [ 3.1259e+01, -3.3500e+01,  1.8257e+01,  ...,  5.8101e+01,\n",
       "             1.7799e+01, -9.7801e+00]],\n",
       "\n",
       "          [[-1.1267e+01,  1.8290e+01, -2.0466e+01,  ...,  1.9381e-01,\n",
       "             1.0076e+01, -1.8775e+01],\n",
       "           [ 4.5241e+00, -1.7194e+01, -1.7535e+01,  ...,  4.2156e+01,\n",
       "             3.4287e+01, -1.8621e+01],\n",
       "           [ 7.6286e-01,  2.3046e+01,  3.7686e+01,  ..., -1.1004e+01,\n",
       "            -6.6670e+01, -1.5847e+01],\n",
       "           ...,\n",
       "           [ 3.7870e+00,  9.1998e+00,  4.9696e+01,  ..., -3.0022e+01,\n",
       "             9.0804e+00, -4.1173e+00],\n",
       "           [ 9.5928e+00, -5.3675e+00, -1.2246e+01,  ..., -1.2674e+01,\n",
       "            -6.3231e+01,  3.7333e+01],\n",
       "           [ 9.7733e+00,  6.2002e+00, -2.8540e+01,  ..., -2.0852e+01,\n",
       "            -1.0783e+01,  4.6889e+00]],\n",
       "\n",
       "          [[-2.5181e+01, -3.0737e+01, -1.1567e+01,  ...,  3.6634e+00,\n",
       "            -4.8850e+00, -2.8742e+01],\n",
       "           [-1.7834e+01, -1.8449e+01,  9.1751e+00,  ..., -2.9554e+01,\n",
       "             4.5491e+01,  1.4389e+01],\n",
       "           [ 5.1467e+01, -2.7407e+00,  3.9518e+01,  ...,  2.1131e+01,\n",
       "            -9.4032e+00, -3.9315e+01],\n",
       "           ...,\n",
       "           [ 8.7318e+00,  2.4657e+01,  6.8517e+00,  ..., -2.2400e+01,\n",
       "             1.8202e+01, -2.3387e+01],\n",
       "           [-2.7750e+01, -4.7269e+00,  5.9178e+00,  ...,  1.6047e+01,\n",
       "             1.5783e+01, -2.1274e+01],\n",
       "           [ 3.2205e+01,  9.4067e+00, -5.5100e+00,  ..., -6.9645e+00,\n",
       "             1.4925e+01,  4.4102e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 5.8826e+00,  2.2175e+01,  1.4812e+01,  ..., -4.3258e+01,\n",
       "             3.5609e+01,  3.0955e+01],\n",
       "           [-4.8723e+00,  3.3080e+00, -3.0947e+01,  ..., -2.3044e+01,\n",
       "            -1.0191e+01,  2.0166e+01],\n",
       "           [ 1.0995e+01, -1.6209e+01,  2.7712e+01,  ...,  4.5249e+01,\n",
       "             2.0264e+01,  2.1453e+01],\n",
       "           ...,\n",
       "           [-2.0640e+01,  3.9230e+00, -1.2991e+01,  ...,  4.2215e+00,\n",
       "            -1.2946e+01, -5.9277e-01],\n",
       "           [ 1.7203e+01,  6.5633e-01, -2.4415e+01,  ..., -3.0588e+01,\n",
       "            -6.3834e+00, -1.9660e+01],\n",
       "           [ 1.7907e+01, -2.1237e+01,  8.1462e+00,  ...,  2.3085e+01,\n",
       "            -1.1617e+01, -6.1826e+00]],\n",
       "\n",
       "          [[-3.1250e+01, -9.0288e+00, -1.7728e+00,  ..., -2.6137e+01,\n",
       "             8.6105e+00,  6.2944e+00],\n",
       "           [-3.6251e+01, -1.0102e+01, -2.0119e+01,  ..., -4.3290e+00,\n",
       "            -1.6395e+01,  2.0992e+01],\n",
       "           [-4.4345e+00,  9.1475e+00, -2.2818e+01,  ...,  2.4301e+01,\n",
       "            -5.6851e+01, -2.2110e+01],\n",
       "           ...,\n",
       "           [ 2.8146e+01,  1.4727e+01, -1.5126e+01,  ..., -1.5019e+01,\n",
       "            -2.7981e+01, -1.4210e+00],\n",
       "           [ 2.8885e+00, -1.2667e+01,  1.3847e+01,  ...,  2.0301e+01,\n",
       "             1.2682e+01,  5.4743e+01],\n",
       "           [-1.1909e+01,  1.0439e+01, -2.4118e+01,  ...,  3.7561e+01,\n",
       "            -1.7115e+01,  2.1718e+01]],\n",
       "\n",
       "          [[-1.3236e+01,  2.8762e+01,  1.0974e+01,  ..., -1.0568e+01,\n",
       "             2.2966e+01,  2.1984e+01],\n",
       "           [-1.5690e+01, -3.1107e+01, -4.6820e+00,  ...,  3.1715e+00,\n",
       "            -6.4360e+00,  1.6634e+01],\n",
       "           [-1.3527e+01, -2.5665e+00,  1.7095e+01,  ..., -5.6784e+01,\n",
       "             9.6765e+00,  3.1421e+00],\n",
       "           ...,\n",
       "           [ 2.0321e+01, -1.0421e+01, -5.4018e+01,  ..., -5.6535e+01,\n",
       "            -2.4972e+01, -2.8788e+01],\n",
       "           [ 1.6033e-01, -3.4517e+01, -8.0431e+00,  ..., -8.5928e+00,\n",
       "             2.8880e+01,  7.9575e+00],\n",
       "           [-9.4592e+00,  3.5541e+01, -9.9372e-01,  ..., -7.7146e+00,\n",
       "            -5.9860e+00,  1.6394e+01]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[ 1.0865e+01, -5.4219e+00, -2.5855e+00,  ...,  2.1135e+01,\n",
       "            -8.9565e+00, -1.1024e+01],\n",
       "           [ 2.0751e+01, -5.3426e+00,  2.5722e-02,  ...,  1.8680e+01,\n",
       "             1.9769e+01, -1.6943e+01],\n",
       "           [-2.2534e+01,  5.4808e+00, -1.3280e-01,  ...,  6.8087e+00,\n",
       "             6.2694e+00, -2.9072e+00],\n",
       "           ...,\n",
       "           [-2.6748e+01,  7.9479e+00,  2.4022e+01,  ..., -1.4677e+01,\n",
       "            -1.4828e+01,  1.1955e+01],\n",
       "           [-6.7372e+00, -2.3523e+01, -6.5896e+01,  ..., -2.1673e+01,\n",
       "            -1.2896e+01, -1.0922e+00],\n",
       "           [ 3.8476e+00, -2.4206e+01,  2.7018e+01,  ...,  5.0001e+00,\n",
       "            -1.2544e-01,  2.8420e+01]],\n",
       "\n",
       "          [[-3.3213e+01,  8.7113e+00, -6.8858e+00,  ..., -4.3560e-01,\n",
       "            -3.7071e+01, -2.3422e+01],\n",
       "           [ 8.0400e+00,  5.2077e+00, -1.2473e+01,  ...,  3.2690e+01,\n",
       "            -9.6345e+00, -4.2642e+01],\n",
       "           [ 2.6811e+01, -2.3517e+01,  3.8737e+01,  ...,  2.4362e+01,\n",
       "             2.2630e+00,  1.3403e+01],\n",
       "           ...,\n",
       "           [-7.0239e+00, -3.7253e+01, -3.2277e+01,  ..., -9.1532e+00,\n",
       "             2.1375e+01, -4.6238e+01],\n",
       "           [ 2.8684e+01, -1.3909e+00,  3.5650e+01,  ...,  1.7248e+01,\n",
       "            -1.7189e+01,  2.6471e+01],\n",
       "           [-1.5310e+01, -6.8954e+00,  2.0791e+00,  ...,  8.1722e+00,\n",
       "             2.6342e+01, -2.4172e+01]],\n",
       "\n",
       "          [[ 9.4904e-01,  2.7225e+01,  2.0420e+01,  ...,  1.5633e+01,\n",
       "            -9.5386e+00, -7.4070e+00],\n",
       "           [-2.8838e+01,  3.3513e+01, -8.3404e+00,  ...,  3.0141e+01,\n",
       "             2.6494e+01,  3.8117e+00],\n",
       "           [-2.0587e+01,  2.2950e+00, -3.2650e+00,  ..., -4.4066e-01,\n",
       "            -2.7066e+01, -2.8529e+01],\n",
       "           ...,\n",
       "           [ 1.0210e+01,  1.1316e+01, -1.5657e+00,  ...,  7.5778e+00,\n",
       "             1.5143e+00,  2.2708e+01],\n",
       "           [-8.6846e+00, -1.3474e+01, -2.1123e+01,  ..., -1.1462e+01,\n",
       "             5.9556e+00, -3.5144e+01],\n",
       "           [-5.1352e+00, -8.5841e+00,  4.2379e+01,  ...,  3.6183e+01,\n",
       "            -1.9681e+01, -9.4644e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.9488e+01,  2.7122e+01,  3.7999e-01,  ..., -9.2137e+00,\n",
       "            -5.2887e+00,  9.2036e+00],\n",
       "           [-2.7818e+01,  1.3298e+01,  8.7349e+00,  ..., -2.3656e+01,\n",
       "             1.3821e+01,  1.3183e+01],\n",
       "           [-1.2520e+00, -2.0606e+01, -6.4970e+00,  ...,  3.6461e+01,\n",
       "            -3.1284e+01, -1.4516e+01],\n",
       "           ...,\n",
       "           [ 6.3518e+00, -8.2607e+00,  1.5571e+01,  ...,  1.0696e+01,\n",
       "            -1.2261e+01,  3.1630e+00],\n",
       "           [-1.5910e+01,  7.1601e+00, -1.6506e-01,  ..., -2.8944e+01,\n",
       "             2.1928e+01,  2.3380e+01],\n",
       "           [ 2.7548e+01,  1.0697e+01, -9.5135e+00,  ..., -3.6714e+01,\n",
       "             5.9809e+00,  9.6850e+00]],\n",
       "\n",
       "          [[-1.2906e+01, -9.1200e+00, -4.6328e+00,  ..., -1.4360e+01,\n",
       "            -1.4250e+00, -6.7076e+00],\n",
       "           [-1.4175e+01, -6.1722e-01,  1.3008e+01,  ...,  9.2713e+00,\n",
       "            -5.4237e+00, -4.3599e+00],\n",
       "           [ 1.0126e+01,  1.9424e+01, -2.2813e+00,  ...,  1.0382e+01,\n",
       "            -2.4770e+01,  1.7228e+00],\n",
       "           ...,\n",
       "           [ 4.5118e+01,  1.5644e+01, -1.6663e+01,  ...,  8.8318e+00,\n",
       "             1.8711e+01,  2.8233e+01],\n",
       "           [-1.0656e+01, -1.1577e+00,  1.3442e+00,  ...,  6.6520e+00,\n",
       "             3.9989e+00, -7.8766e+00],\n",
       "           [-1.1505e+01, -2.5947e+01, -2.6891e+00,  ..., -3.7987e+01,\n",
       "             1.0177e+01,  1.6664e+01]],\n",
       "\n",
       "          [[ 7.0857e+00, -1.2619e+01, -3.2071e+01,  ..., -4.8682e+01,\n",
       "            -7.4331e+00, -2.2477e+01],\n",
       "           [-1.1480e+01, -2.7963e+01,  2.8021e+01,  ...,  1.0845e+01,\n",
       "             2.9183e+01, -4.9454e+00],\n",
       "           [-1.9186e+01, -1.6635e+01, -6.6415e+00,  ...,  7.8366e+00,\n",
       "             2.6700e+01, -8.3859e+00],\n",
       "           ...,\n",
       "           [-1.4511e+01, -1.0513e+01, -1.3825e+01,  ..., -1.2995e+01,\n",
       "            -1.5554e+01, -1.9079e+01],\n",
       "           [-5.1422e+00, -2.4481e+01, -9.6857e+00,  ..., -6.8114e+00,\n",
       "            -8.5567e+00,  1.2220e+01],\n",
       "           [-4.8555e+01,  1.4572e+01,  1.7624e+01,  ...,  8.2315e+00,\n",
       "            -9.8546e+00,  4.6997e+00]]],\n",
       "\n",
       "\n",
       "         [[[ 4.1902e+00,  1.0483e+01, -9.3523e+00,  ...,  3.7505e+01,\n",
       "            -2.7745e+00, -6.5890e+00],\n",
       "           [ 1.4784e+01,  8.0337e+00,  1.4866e+01,  ..., -7.1198e+00,\n",
       "            -9.3535e+00, -3.0763e+00],\n",
       "           [ 1.5292e+00,  5.6408e+00,  1.7512e+01,  ...,  1.7638e+01,\n",
       "             2.1300e+01,  2.5713e+01],\n",
       "           ...,\n",
       "           [ 9.6130e+00,  1.3785e+01, -7.0409e+00,  ...,  8.7471e+00,\n",
       "             1.8576e+01, -1.8022e+01],\n",
       "           [ 1.1407e+00,  1.2249e+01,  3.2682e+00,  ..., -2.8517e+01,\n",
       "            -9.6228e+00, -3.9968e+01],\n",
       "           [ 3.7751e+00,  9.7817e+00, -6.6411e+00,  ...,  2.3353e+01,\n",
       "            -7.9435e+00,  1.7332e+01]],\n",
       "\n",
       "          [[ 3.2098e+01,  5.4307e+00, -2.2401e+01,  ...,  7.3728e+00,\n",
       "             4.3592e+00,  1.7563e+01],\n",
       "           [-8.3859e-01, -1.5297e+00, -3.5103e+01,  ...,  7.0990e+00,\n",
       "            -2.0410e+01, -2.2721e+00],\n",
       "           [ 3.9612e+00,  5.1148e+00,  4.0558e+00,  ..., -1.4123e+01,\n",
       "            -8.3190e+00,  2.1101e+01],\n",
       "           ...,\n",
       "           [-1.8724e+00, -5.8293e+00, -8.6479e+00,  ...,  1.1880e+01,\n",
       "             3.0079e+00, -1.2892e+01],\n",
       "           [-3.2161e+01,  3.2144e+01,  4.0818e+01,  ..., -1.1995e+01,\n",
       "             3.7954e+00, -6.3030e+00],\n",
       "           [ 2.7683e+01,  2.3730e+01, -8.0587e+00,  ...,  4.2975e+00,\n",
       "             9.4748e-01, -9.0353e+00]],\n",
       "\n",
       "          [[ 2.2030e+01, -5.9223e+00,  2.9539e+01,  ..., -4.8439e+00,\n",
       "            -2.1275e+01, -2.6157e+01],\n",
       "           [ 2.1999e+01, -1.8650e+01, -2.2235e+00,  ...,  3.4534e+01,\n",
       "            -4.1151e+00,  3.5136e+01],\n",
       "           [ 5.3725e+00, -1.7236e+00,  6.4020e+00,  ..., -4.6464e+01,\n",
       "             4.5776e-04, -4.0366e+01],\n",
       "           ...,\n",
       "           [-1.3078e+01, -1.8343e+01,  1.3716e+00,  ...,  2.5563e+00,\n",
       "            -3.9833e+01, -5.5871e+00],\n",
       "           [ 1.3586e+01, -9.7061e+00,  6.9463e+00,  ..., -7.7831e+00,\n",
       "            -2.2562e+01,  3.8459e+00],\n",
       "           [ 2.5019e+01, -1.2332e+00,  9.4637e+00,  ...,  3.8989e+01,\n",
       "             5.5866e+00,  3.9535e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-8.2924e+00,  1.1108e+01, -6.5400e+00,  ..., -3.5771e+01,\n",
       "             1.4677e+01, -1.1009e+01],\n",
       "           [ 2.5050e+01, -2.5986e+01,  7.6214e+00,  ..., -8.2667e+00,\n",
       "            -5.0382e+01,  3.2983e+00],\n",
       "           [-1.1218e+00, -8.7721e+00, -9.1794e+00,  ...,  2.1423e+01,\n",
       "             3.4853e+01, -2.7809e+01],\n",
       "           ...,\n",
       "           [ 9.4451e+00,  9.5802e+00,  1.4858e+01,  ..., -3.5967e+01,\n",
       "             3.5920e+01, -9.8569e+00],\n",
       "           [-6.4609e+00, -1.2249e+01, -1.6708e-01,  ...,  6.4339e+00,\n",
       "            -1.1458e+01, -7.2350e+00],\n",
       "           [ 5.9573e+00,  5.7535e+00, -2.8736e+01,  ..., -1.4531e+01,\n",
       "            -3.1936e+01,  1.3989e+01]],\n",
       "\n",
       "          [[ 1.9234e+01,  4.2586e+00,  3.1623e+01,  ..., -4.2464e+01,\n",
       "             2.1746e+01,  3.4776e+01],\n",
       "           [ 4.9834e+00, -5.6349e+00, -8.9505e+00,  ..., -1.9076e+01,\n",
       "             3.7571e+01, -1.0802e+01],\n",
       "           [-1.0505e+01, -2.2615e+01, -1.2247e+01,  ...,  3.9070e+01,\n",
       "            -6.9294e+00,  1.8999e+01],\n",
       "           ...,\n",
       "           [ 9.2868e+00, -2.2466e+01, -8.9614e+00,  ..., -2.9119e+01,\n",
       "            -2.6690e+01,  3.6360e+01],\n",
       "           [ 2.8281e+00,  1.9682e+01, -1.2495e+01,  ...,  1.4123e+01,\n",
       "             3.8391e+01, -4.2228e+01],\n",
       "           [-3.4929e+01,  2.4123e+01, -2.5312e+01,  ..., -1.6362e+01,\n",
       "             3.8119e+01,  1.9848e+01]],\n",
       "\n",
       "          [[-9.1725e+00,  1.9027e+01,  4.1454e+01,  ..., -5.6455e+00,\n",
       "             5.1432e+01, -1.8804e+01],\n",
       "           [ 9.7141e+00,  3.0041e+00,  2.2616e+01,  ...,  2.0355e-01,\n",
       "            -1.0861e+01,  5.7830e+00],\n",
       "           [-2.8515e+01,  4.5114e+00, -2.5703e+01,  ...,  2.2157e+01,\n",
       "            -2.4580e+00, -2.8460e+00],\n",
       "           ...,\n",
       "           [ 1.7878e+01,  2.9661e+01,  8.8819e+00,  ..., -1.5780e+01,\n",
       "             2.0994e+01, -4.4845e+00],\n",
       "           [ 3.4867e+01,  1.3837e+01, -1.6992e+01,  ..., -3.7539e+01,\n",
       "            -4.5608e+01,  1.7844e+01],\n",
       "           [-3.4602e+01, -6.6867e+00,  2.3663e+01,  ...,  1.2567e+01,\n",
       "            -1.8657e+00,  2.0744e+01]]],\n",
       "\n",
       "\n",
       "         [[[-1.1768e+00, -1.6930e+01,  1.1646e+01,  ...,  1.2332e+01,\n",
       "             1.3063e+01,  7.3927e+00],\n",
       "           [ 3.6670e+00,  9.9658e+00,  2.4059e+01,  ..., -2.6544e+01,\n",
       "             1.2984e+01, -2.9974e+01],\n",
       "           [-2.3434e+01, -3.7064e+01,  1.4167e+01,  ..., -3.5676e+01,\n",
       "            -2.9789e+01, -1.6418e+01],\n",
       "           ...,\n",
       "           [ 8.4938e+00, -1.3436e+01, -1.8190e+01,  ...,  4.5025e+01,\n",
       "             4.3090e+01,  1.9364e+00],\n",
       "           [ 1.0259e+01, -2.0054e+01, -3.7964e-01,  ...,  1.4681e+01,\n",
       "            -2.3976e+01,  9.5882e-01],\n",
       "           [-2.6501e+01,  4.7645e+01, -2.0561e+01,  ...,  5.9927e+00,\n",
       "            -2.0622e+01,  5.8128e+01]],\n",
       "\n",
       "          [[ 1.8172e+01,  3.5754e+01, -3.1327e+01,  ...,  1.9052e+00,\n",
       "             2.1168e+01,  9.5336e+00],\n",
       "           [-2.5688e+01, -1.3846e+01, -2.0286e+01,  ...,  1.5211e+01,\n",
       "            -2.9720e+01, -3.8450e+01],\n",
       "           [-2.8514e+01,  2.2489e+00,  8.5543e+00,  ...,  3.5857e+01,\n",
       "             7.1941e+00, -9.1236e+00],\n",
       "           ...,\n",
       "           [ 1.8414e+01, -9.9601e+00, -1.9420e+01,  ..., -1.5032e+01,\n",
       "            -7.3701e+00, -2.9072e+01],\n",
       "           [ 2.2878e+01, -4.4370e+01, -6.6500e+00,  ..., -3.3820e+00,\n",
       "            -7.4409e+00,  9.8086e+00],\n",
       "           [ 1.5358e+01,  2.1960e+00,  1.2291e+01,  ..., -3.2974e+01,\n",
       "             1.1952e+01, -4.9207e+00]],\n",
       "\n",
       "          [[-2.1986e+01, -3.6484e+01,  4.2217e+01,  ...,  2.2247e+01,\n",
       "            -1.2109e+01,  1.6271e+01],\n",
       "           [ 1.7600e+01,  1.9944e+01,  5.8801e+00,  ...,  7.3711e+00,\n",
       "             1.8358e+01, -7.1644e+00],\n",
       "           [-5.1815e+00, -1.8100e+01, -4.0089e+01,  ..., -1.4106e+01,\n",
       "            -8.0720e+00, -3.4160e+01],\n",
       "           ...,\n",
       "           [ 9.1363e+00, -2.0252e+00, -1.3911e+00,  ..., -5.7914e+00,\n",
       "            -4.7335e+00, -5.6697e+00],\n",
       "           [-3.4701e+00, -2.9712e-01, -2.0678e+01,  ...,  9.6714e+00,\n",
       "            -2.2256e+01,  2.9324e+01],\n",
       "           [-9.1024e+00,  1.5777e+01, -1.8474e+01,  ..., -2.4648e+01,\n",
       "             9.9511e+00,  5.1798e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.1065e+01,  1.8686e+01, -6.7392e+00,  ...,  3.4302e+01,\n",
       "            -2.8225e+01,  2.6373e-01],\n",
       "           [ 2.1100e+01, -2.8942e+00, -3.6518e+01,  ...,  2.8060e+00,\n",
       "            -1.4897e+01,  1.5511e+01],\n",
       "           [-2.4041e+01,  2.7905e+00,  4.9506e+00,  ..., -9.2302e+00,\n",
       "             2.8022e+01, -2.1233e+01],\n",
       "           ...,\n",
       "           [ 3.5210e-01,  1.6527e+01, -2.1986e+01,  ..., -3.9390e+01,\n",
       "            -1.1902e+01, -2.3479e+01],\n",
       "           [ 2.3782e+01, -1.8052e+01, -6.5071e+00,  ..., -1.9559e+01,\n",
       "             1.0733e+01,  1.4079e+01],\n",
       "           [-1.2720e+01, -2.1163e+01,  2.5666e+00,  ...,  1.8704e+01,\n",
       "             1.1273e+01, -1.6888e+01]],\n",
       "\n",
       "          [[ 3.3917e+01,  1.2318e+01, -1.6248e+01,  ...,  2.2730e+01,\n",
       "             1.8615e+01, -3.2489e+01],\n",
       "           [ 5.3522e+00,  9.2716e+00, -7.2456e+00,  ..., -2.5566e+00,\n",
       "             8.1131e+00, -3.1509e+01],\n",
       "           [ 5.1467e+00,  6.1579e+01, -2.1677e+01,  ..., -3.3901e+00,\n",
       "            -2.8751e+00,  3.6941e+00],\n",
       "           ...,\n",
       "           [ 2.9626e+01, -4.8905e+00,  4.0993e+00,  ...,  1.3611e+01,\n",
       "             4.4492e+00,  3.0236e+00],\n",
       "           [-1.5660e+01, -2.8173e+00,  1.7847e+00,  ...,  2.9399e+00,\n",
       "             2.1539e+01,  1.3572e+01],\n",
       "           [-7.2025e-01, -1.3552e+01,  1.2395e+01,  ...,  7.4171e+00,\n",
       "             1.5187e+01, -8.8394e+00]],\n",
       "\n",
       "          [[-5.4095e+00,  6.5875e+00,  3.7017e+00,  ...,  2.6544e+01,\n",
       "            -3.2811e+01,  2.5931e+01],\n",
       "           [-4.3458e+01, -3.4330e+01, -1.4262e+01,  ..., -2.6371e+01,\n",
       "            -9.2157e+00, -4.5994e+01],\n",
       "           [ 1.8751e+01,  3.6634e+01, -3.6518e+01,  ...,  1.5348e-01,\n",
       "             3.0133e+01, -4.6055e+01],\n",
       "           ...,\n",
       "           [ 3.7827e+01, -6.3621e+00, -1.6220e+01,  ..., -1.3657e+01,\n",
       "            -2.0680e+01,  3.5857e+00],\n",
       "           [ 2.3391e+00,  3.9539e+00, -6.5428e+00,  ..., -2.3790e+01,\n",
       "             7.0730e+00,  4.0658e+01],\n",
       "           [ 2.9898e+01,  1.4176e+01,  1.6598e+01,  ..., -1.0430e+01,\n",
       "             6.0404e+00, -6.2248e+00]]]],\n",
       "\n",
       "\n",
       "\n",
       "        [[[[-2.4525e+00,  3.2384e+01, -1.6479e+01,  ...,  2.1939e+01,\n",
       "             3.1133e+01, -4.3714e+01],\n",
       "           [ 5.0838e-01,  4.2107e+00,  1.3324e+01,  ...,  1.6129e+01,\n",
       "             4.6229e+00,  3.5869e+01],\n",
       "           [-3.4134e+01, -1.0277e+01, -4.0637e+01,  ..., -2.1477e+01,\n",
       "             5.6412e+00, -1.1588e+00],\n",
       "           ...,\n",
       "           [-1.2021e+01,  3.5733e+01, -2.2558e+01,  ...,  4.8655e+00,\n",
       "            -7.3706e+00,  1.7419e+01],\n",
       "           [-6.7425e+00,  2.3313e+01, -2.5579e+01,  ..., -4.3222e+01,\n",
       "             2.4560e+00,  7.3328e+00],\n",
       "           [-5.9835e+00, -4.4217e+00,  5.0676e+01,  ..., -3.5036e+00,\n",
       "            -7.5274e+00, -2.3268e+01]],\n",
       "\n",
       "          [[ 7.6078e+00, -2.4559e+01, -3.4460e+01,  ..., -3.9554e+00,\n",
       "            -1.6767e+01, -4.3030e+00],\n",
       "           [-8.5586e+00,  1.5927e+01,  5.1307e+00,  ...,  9.9067e-01,\n",
       "            -3.4999e+01, -5.9867e+00],\n",
       "           [ 2.7917e+00, -7.8433e+00,  4.0179e+00,  ...,  9.7567e+00,\n",
       "            -3.8881e+01,  1.3945e+00],\n",
       "           ...,\n",
       "           [ 2.2588e+01,  1.6397e+01,  3.2725e+01,  ..., -1.9924e+00,\n",
       "             1.9999e+01,  4.2948e+00],\n",
       "           [-3.1659e+01,  7.9218e+00, -1.4597e+01,  ..., -2.3077e+01,\n",
       "            -2.0643e+01, -1.2408e+01],\n",
       "           [-8.6188e-01, -3.1823e+01, -3.4173e+00,  ..., -1.2829e+00,\n",
       "            -8.7394e+00, -3.6583e+00]],\n",
       "\n",
       "          [[-7.6702e-01,  9.5238e-01, -1.6064e+01,  ...,  8.5283e+00,\n",
       "            -4.9903e+01,  8.5789e-01],\n",
       "           [ 3.6357e+01, -2.4550e+01, -5.2524e+00,  ..., -2.6880e+01,\n",
       "             4.1852e+00, -5.2398e+00],\n",
       "           [ 4.6049e+01, -2.2561e+01,  1.2873e+01,  ...,  5.1813e+00,\n",
       "            -2.7749e+01, -2.2880e+01],\n",
       "           ...,\n",
       "           [ 2.8465e+01, -5.7354e+01, -5.2058e+01,  ...,  2.8259e+01,\n",
       "             1.7564e+01,  5.9395e+00],\n",
       "           [-5.6460e+00, -1.7173e+01, -2.2120e+01,  ..., -1.6201e+01,\n",
       "             3.7073e+01,  1.1520e+01],\n",
       "           [-2.3150e+01, -1.8472e+01, -3.5557e+01,  ...,  1.9282e+01,\n",
       "            -3.4276e+00,  7.3446e-01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 7.5804e+00,  9.2036e+00, -8.3347e+00,  ..., -1.6269e+01,\n",
       "             1.6988e+01,  7.1290e+01],\n",
       "           [-3.0122e+01, -1.2907e+01,  1.7050e+01,  ..., -3.3052e+01,\n",
       "             1.5827e+01,  1.1318e+01],\n",
       "           [-3.3530e+01,  1.8157e+00,  2.6847e+01,  ..., -1.3754e+01,\n",
       "             1.1423e+01, -3.7732e+01],\n",
       "           ...,\n",
       "           [ 2.2772e+01,  1.9140e+01,  1.4298e+01,  ...,  6.6249e+00,\n",
       "            -4.8406e+00,  8.9640e+00],\n",
       "           [ 1.9273e+01, -7.4005e+00,  3.2768e+01,  ..., -1.7335e+01,\n",
       "             8.8747e+00, -1.6945e+01],\n",
       "           [-2.4366e+01,  4.6804e+00,  4.4333e+01,  ...,  7.6541e+00,\n",
       "             1.9888e+01,  2.4868e+01]],\n",
       "\n",
       "          [[ 5.9264e+00,  7.5227e+00,  1.5068e+01,  ...,  2.2745e+01,\n",
       "             1.6347e+01,  1.9232e+01],\n",
       "           [-3.6804e+00, -9.4755e+00,  4.9898e+00,  ..., -1.6179e+01,\n",
       "             1.2876e+01, -1.0368e+01],\n",
       "           [ 1.1398e+00, -8.9081e+00,  2.4876e+01,  ..., -2.1047e+01,\n",
       "             2.3663e+01,  8.0146e+00],\n",
       "           ...,\n",
       "           [-1.9553e+01, -1.3683e+00,  7.9274e+00,  ...,  2.5352e+01,\n",
       "             5.5960e+00,  1.4176e+01],\n",
       "           [ 1.4241e+01,  3.9424e+01,  1.0744e+00,  ...,  2.0842e+01,\n",
       "             2.7639e+01,  6.8420e+00],\n",
       "           [-4.6613e+00, -5.6768e-01, -1.4284e+01,  ...,  3.0803e+01,\n",
       "            -1.4858e+01, -2.9725e+00]],\n",
       "\n",
       "          [[-2.0328e+01, -2.1226e+01,  2.8049e+01,  ...,  2.8601e+00,\n",
       "             2.1445e+00, -3.6660e+01],\n",
       "           [-5.7556e+01,  4.2630e+00, -3.4819e+01,  ...,  9.4612e+00,\n",
       "             1.8690e+01, -2.4138e+00],\n",
       "           [ 2.6063e+00,  3.1101e+01, -3.5715e+00,  ...,  1.9960e+00,\n",
       "             5.5052e+00,  1.4289e+01],\n",
       "           ...,\n",
       "           [-2.9585e+01,  7.2910e+00, -1.7177e+00,  ..., -1.8242e+01,\n",
       "            -1.9566e+00, -6.6463e+00],\n",
       "           [-1.9914e+01, -1.5287e+01,  4.4548e+01,  ...,  3.3766e+01,\n",
       "             1.8983e+01,  2.1065e+01],\n",
       "           [ 2.6105e+00,  3.0537e+01, -4.7602e+01,  ...,  4.7289e+01,\n",
       "             1.8752e+01,  8.1189e+00]]],\n",
       "\n",
       "\n",
       "         [[[ 1.4190e+01, -1.9746e+01,  6.3716e-01,  ...,  1.1646e+01,\n",
       "            -1.0378e+00,  1.3909e+01],\n",
       "           [ 2.7996e+01,  1.5461e+01,  7.6292e-01,  ...,  1.0651e+01,\n",
       "            -1.6864e+01, -2.6918e+01],\n",
       "           [ 4.9674e+01, -3.7372e+01,  2.2594e+00,  ..., -1.8617e+01,\n",
       "             1.1748e+01, -3.1065e+01],\n",
       "           ...,\n",
       "           [-3.4314e+00,  2.5631e+01,  1.7822e+01,  ...,  2.0317e+01,\n",
       "            -1.0557e+01,  1.7911e+01],\n",
       "           [ 1.2294e+01, -1.0332e+01, -3.5548e+00,  ..., -1.8104e+01,\n",
       "            -9.7398e+00, -1.0478e+01],\n",
       "           [-9.0252e+00,  8.2296e-01,  1.9121e+00,  ..., -1.9150e+01,\n",
       "             3.7684e+01, -2.0289e+01]],\n",
       "\n",
       "          [[ 4.9266e+00, -3.0719e+01, -2.9299e+01,  ..., -1.5773e+01,\n",
       "            -1.8894e+00, -1.0112e+01],\n",
       "           [-5.3488e+01, -3.8452e+00, -4.3131e+01,  ..., -1.1750e-01,\n",
       "             3.5027e+00, -8.9623e+00],\n",
       "           [-1.9958e+01, -3.0268e+00,  4.7099e+00,  ..., -1.6549e+00,\n",
       "             1.6862e+01, -1.5276e+01],\n",
       "           ...,\n",
       "           [-1.6155e+01,  1.3844e+01,  4.6498e-01,  ...,  5.3027e+01,\n",
       "            -5.5473e+00,  5.7307e+00],\n",
       "           [ 2.1341e+01, -1.3523e+01, -1.6483e+01,  ..., -1.9702e+01,\n",
       "            -1.4547e+01,  1.0431e+01],\n",
       "           [ 9.2546e+00,  3.0011e+01,  3.2327e+01,  ..., -2.6399e+01,\n",
       "             3.0801e+01, -2.5313e+01]],\n",
       "\n",
       "          [[-3.0169e+01,  1.0362e+01,  3.5765e+01,  ...,  2.1363e+01,\n",
       "             2.0983e+01, -2.7816e+01],\n",
       "           [ 1.6106e+01,  3.4361e+01, -1.6074e+01,  ...,  1.9075e+01,\n",
       "             3.7105e+01, -2.9953e+00],\n",
       "           [-2.2667e+01,  2.2679e+01, -2.1755e+01,  ..., -9.4245e-01,\n",
       "             3.4704e+01, -7.5207e+00],\n",
       "           ...,\n",
       "           [-2.8571e+01, -1.0907e+01,  2.9456e+01,  ...,  9.6001e+00,\n",
       "            -4.0413e+00, -6.3357e+00],\n",
       "           [ 3.8734e+01, -2.1757e+00,  8.7910e+00,  ...,  2.6959e+01,\n",
       "             3.8991e+01,  3.1472e+01],\n",
       "           [-6.6498e+00,  9.4714e+00, -1.3198e+00,  ...,  3.8667e+00,\n",
       "             2.7875e+00, -5.6048e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.3618e+01,  4.3691e+01,  5.3522e-01,  ...,  5.7441e+00,\n",
       "            -1.1237e+01, -6.2523e+00],\n",
       "           [ 1.9776e+01,  1.2341e+01, -4.3111e+01,  ...,  4.2006e+00,\n",
       "            -1.9176e+01,  2.5171e+01],\n",
       "           [ 1.5179e+01, -1.1375e+01, -1.7118e+01,  ..., -5.9944e+00,\n",
       "             1.6843e+01,  8.8664e+00],\n",
       "           ...,\n",
       "           [ 2.9612e+00, -2.7351e+01, -1.1174e+01,  ..., -2.3659e+01,\n",
       "            -1.2929e+01, -9.7848e+00],\n",
       "           [ 1.9075e+01,  1.1671e+01, -1.9348e+01,  ...,  1.0869e+01,\n",
       "             3.5205e+01, -2.9603e+01],\n",
       "           [ 4.6584e+00,  2.9750e+00, -5.7820e+00,  ..., -4.0825e+00,\n",
       "            -5.3215e+00, -9.5776e+00]],\n",
       "\n",
       "          [[-3.2581e+00,  9.8770e+00, -2.3772e+01,  ...,  1.6025e+01,\n",
       "             9.2253e+00, -6.9365e+00],\n",
       "           [-1.2165e+01,  3.6472e+00,  2.6171e+01,  ...,  2.2280e+01,\n",
       "            -2.6007e+01,  5.1769e-02],\n",
       "           [-4.6877e+00,  4.1793e+01, -3.5760e+01,  ...,  6.8514e+00,\n",
       "            -2.7644e+01,  2.5275e+01],\n",
       "           ...,\n",
       "           [ 6.8406e-01,  1.3309e+01,  1.3443e+01,  ..., -1.5683e+01,\n",
       "            -2.5352e+01, -3.4748e+01],\n",
       "           [-1.0120e+00, -2.3747e+01,  2.0322e+00,  ..., -6.1139e+01,\n",
       "            -5.6575e+01, -1.2550e+01],\n",
       "           [-1.0783e+01,  6.3524e+00,  1.3739e+00,  ..., -4.8235e+00,\n",
       "             3.0658e+01, -9.8048e+00]],\n",
       "\n",
       "          [[-8.9134e+00,  1.2619e+01,  3.3314e+00,  ..., -3.6582e+01,\n",
       "            -2.8145e+00,  9.6723e+00],\n",
       "           [ 2.9132e+01, -4.5034e+01, -1.9409e+01,  ...,  2.1922e+01,\n",
       "             5.1904e+00, -1.3851e+01],\n",
       "           [-4.2845e+01, -1.1743e+01,  1.3416e+01,  ...,  1.7512e+01,\n",
       "             8.8463e+00,  5.0706e+00],\n",
       "           ...,\n",
       "           [-1.3923e+01, -1.5043e+01,  1.9277e+01,  ..., -4.4321e+00,\n",
       "            -3.9413e+01, -3.5512e+01],\n",
       "           [-1.4449e+01,  1.9280e+00, -1.9551e+01,  ...,  5.4257e+00,\n",
       "            -1.7018e+01,  7.1583e+00],\n",
       "           [-3.4703e+00, -3.4401e-01, -5.8301e+00,  ...,  3.1077e+01,\n",
       "             1.0148e+01, -1.0666e+01]]],\n",
       "\n",
       "\n",
       "         [[[-3.1645e+00, -4.8271e+00,  4.7525e+00,  ...,  7.7404e+00,\n",
       "            -9.6234e+00,  2.0221e+01],\n",
       "           [-2.0430e+01,  2.3068e+00,  2.5208e+01,  ...,  1.3370e+01,\n",
       "             2.5536e+01,  3.2298e+01],\n",
       "           [-1.1861e+01,  2.2631e+01,  1.1449e+00,  ...,  4.7619e+01,\n",
       "            -3.2341e+01,  3.3091e+01],\n",
       "           ...,\n",
       "           [-2.2678e+01, -9.0177e-01, -7.6145e+00,  ...,  1.9840e+01,\n",
       "            -1.5232e+01,  1.8316e+01],\n",
       "           [ 3.8736e+00, -1.0357e+01,  4.4126e+01,  ..., -6.3023e+00,\n",
       "            -3.4681e+01,  2.1252e+01],\n",
       "           [-4.7046e+00, -1.9601e+01, -8.5011e-03,  ...,  5.5520e+01,\n",
       "             7.1789e+00,  7.8727e+00]],\n",
       "\n",
       "          [[-4.5916e+00, -4.5372e+00, -2.5207e+00,  ..., -6.4675e+00,\n",
       "            -1.3409e+01, -6.5810e+00],\n",
       "           [ 1.0368e+01, -2.0615e+01,  1.1776e+01,  ..., -2.4552e+01,\n",
       "            -4.3931e+00, -1.7600e+01],\n",
       "           [ 8.0366e+00, -1.9564e+01,  3.9428e+01,  ...,  1.5176e+01,\n",
       "            -1.9897e+01,  3.1394e+01],\n",
       "           ...,\n",
       "           [-1.1922e+01, -2.9977e+01, -1.5983e+01,  ..., -1.0038e+01,\n",
       "             2.4292e+01, -1.7701e+00],\n",
       "           [ 1.9573e+01,  3.3566e+01,  3.0734e+00,  ..., -4.2120e+01,\n",
       "            -6.6925e+00, -3.5101e+00],\n",
       "           [ 5.5999e+00,  4.6612e+00,  3.7404e+01,  ...,  7.1312e+00,\n",
       "             3.4150e+01, -2.2798e+00]],\n",
       "\n",
       "          [[-3.8573e+01,  1.4428e+01, -4.8753e+01,  ...,  1.0701e+01,\n",
       "             4.0522e+01, -1.0439e+01],\n",
       "           [-2.5127e+01, -2.6701e+01, -7.5713e+00,  ..., -3.7970e+01,\n",
       "             2.8943e+01, -3.0383e+00],\n",
       "           [-6.5771e+00, -1.9012e+01, -7.6579e+00,  ..., -1.2441e+01,\n",
       "            -1.3167e+01, -3.0779e+01],\n",
       "           ...,\n",
       "           [-3.8965e+00,  2.0002e+01, -4.8101e+01,  ..., -3.3748e+01,\n",
       "            -8.1597e+00,  6.5464e+00],\n",
       "           [ 9.7298e+00,  6.9409e+00,  1.3979e+01,  ..., -2.7322e+01,\n",
       "             8.5783e-01, -1.2029e+01],\n",
       "           [ 8.9871e+00,  3.2697e+01, -3.5450e+01,  ..., -1.3993e+01,\n",
       "             2.1626e+01, -6.4623e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.8088e+01, -9.8254e+00, -5.8253e+00,  ...,  5.9673e+00,\n",
       "             1.6165e+01,  2.5957e+01],\n",
       "           [-4.9947e+01,  4.1200e+01,  3.3324e+01,  ..., -1.8836e+01,\n",
       "            -1.7645e+01,  2.8503e+01],\n",
       "           [-1.5464e+01, -1.8824e+01,  1.0945e+01,  ..., -2.4896e+00,\n",
       "             2.6033e+01, -2.7323e+01],\n",
       "           ...,\n",
       "           [ 2.4133e+01,  1.2610e+01,  3.4000e+01,  ..., -4.1460e+00,\n",
       "             1.3556e+01, -1.5839e+01],\n",
       "           [-2.7508e+01, -5.4211e+00,  9.9409e-01,  ..., -7.0398e-01,\n",
       "             2.9038e-01, -5.4238e+01],\n",
       "           [ 6.9263e+00,  4.0436e+00,  3.2984e+01,  ...,  1.6803e+00,\n",
       "            -3.8783e+01, -4.8261e+01]],\n",
       "\n",
       "          [[-4.6557e+01, -5.0090e+00, -9.9622e+00,  ..., -1.3917e+01,\n",
       "             2.3115e+01, -4.9854e+00],\n",
       "           [ 1.6786e+01,  5.2384e+00,  1.3004e+01,  ...,  8.4520e+00,\n",
       "            -3.2068e+00,  2.3690e+01],\n",
       "           [ 1.7948e+01,  8.0539e+00, -4.5868e-01,  ...,  1.5346e+01,\n",
       "            -9.3602e+00, -1.1684e+01],\n",
       "           ...,\n",
       "           [-3.0402e+01,  1.3524e+01, -1.7419e+00,  ...,  1.1848e+01,\n",
       "             5.8109e+01,  4.0558e+01],\n",
       "           [ 5.0147e+01, -1.4299e+01,  7.4478e+00,  ..., -8.7090e+00,\n",
       "             1.0498e+00, -1.0328e+01],\n",
       "           [ 1.5015e+01,  2.8606e+01, -2.3080e+01,  ..., -1.1720e+00,\n",
       "            -8.4294e+00, -3.6077e+01]],\n",
       "\n",
       "          [[ 5.5216e-01, -1.2988e+00,  1.2504e+01,  ..., -5.9116e-01,\n",
       "            -3.3261e+01,  1.4912e+01],\n",
       "           [ 9.8860e+00, -6.2263e+00,  5.2651e+00,  ...,  4.9998e+01,\n",
       "             9.5327e+00,  3.0340e+01],\n",
       "           [ 7.9342e+00, -6.5499e+00,  2.6041e+01,  ..., -5.6620e+00,\n",
       "             1.4193e+01, -5.4723e+00],\n",
       "           ...,\n",
       "           [-1.2591e+01,  1.8723e+01, -2.2521e+00,  ...,  7.3932e+00,\n",
       "             6.1280e+00,  3.5087e+01],\n",
       "           [ 7.6550e-01,  3.8366e+01,  1.3451e+00,  ..., -1.7065e+01,\n",
       "             3.5957e-02, -1.0295e+01],\n",
       "           [-1.3773e+00, -8.2444e+00, -6.0033e+01,  ...,  5.1505e+01,\n",
       "            -3.9539e+01,  6.8462e+00]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[-2.5017e+01, -5.5112e+00,  1.2072e+01,  ...,  1.1858e+01,\n",
       "             1.5614e+01,  2.8958e+01],\n",
       "           [-1.3334e+01, -2.1740e+01, -9.5124e+00,  ..., -3.1125e+01,\n",
       "            -1.0323e+01, -4.2172e+00],\n",
       "           [-2.9147e+01,  2.6112e+01, -7.1798e+00,  ...,  1.9567e+01,\n",
       "             1.9210e+01,  2.1511e+01],\n",
       "           ...,\n",
       "           [ 2.0318e+01, -1.3131e+01,  2.6549e+01,  ..., -1.9516e+01,\n",
       "            -1.5146e+00,  1.7336e+01],\n",
       "           [-2.6179e+01,  3.2017e+01,  8.6804e+00,  ...,  1.8842e+01,\n",
       "            -6.5865e+00,  9.8102e+00],\n",
       "           [-2.0114e+01, -4.9378e+01, -1.9108e+01,  ..., -2.8010e+01,\n",
       "            -1.0270e+01,  1.1626e+01]],\n",
       "\n",
       "          [[-5.3007e+00,  1.1484e+01,  4.4348e+00,  ..., -1.9156e+00,\n",
       "            -1.7453e+01, -3.0912e+01],\n",
       "           [-3.7384e+01,  1.5705e+01, -1.6678e+01,  ..., -5.1015e+01,\n",
       "             2.0958e+01,  1.2469e+01],\n",
       "           [-1.9491e+00,  5.3472e+00,  2.1344e+01,  ...,  2.9006e+01,\n",
       "            -4.5471e+00, -2.4003e+01],\n",
       "           ...,\n",
       "           [-4.0008e+00,  3.4063e-02,  1.1624e+01,  ..., -2.8023e+01,\n",
       "             2.4236e+01, -1.2213e+01],\n",
       "           [-3.9602e+00,  1.2349e+01,  2.5934e+01,  ..., -1.4267e+01,\n",
       "             2.6156e+01, -1.2866e+01],\n",
       "           [-2.1228e+00, -9.7126e+00,  1.8880e+01,  ..., -1.1088e+01,\n",
       "             3.1086e+01, -1.1327e+01]],\n",
       "\n",
       "          [[-8.9845e+00, -2.0646e+01,  6.5038e+00,  ...,  2.7799e+00,\n",
       "            -1.1146e+01, -6.1420e+00],\n",
       "           [-1.6885e+01, -4.7112e+00,  4.2138e+00,  ...,  9.8507e+00,\n",
       "            -5.4079e+00, -1.2720e+00],\n",
       "           [ 1.4297e+01,  1.3147e+01, -1.1207e+01,  ..., -3.0900e+00,\n",
       "             2.5113e+00,  9.8766e+00],\n",
       "           ...,\n",
       "           [ 1.4539e+01, -2.6955e+01,  2.1274e+01,  ..., -2.2828e+01,\n",
       "             5.1920e+00, -2.9266e+01],\n",
       "           [ 7.4194e+00,  6.3763e+00, -5.2840e+00,  ...,  5.6563e+00,\n",
       "            -2.7622e+01, -3.2141e+01],\n",
       "           [-2.3338e+00,  1.9024e+01,  1.2994e+01,  ...,  1.7597e+01,\n",
       "             9.0456e+00, -1.2095e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 3.6484e+00,  2.2951e+01, -1.4081e+01,  ..., -2.0672e+01,\n",
       "             5.4679e+01, -1.5299e+01],\n",
       "           [ 2.0310e+01, -8.9618e+00,  7.3711e+00,  ..., -3.5179e+01,\n",
       "            -8.3769e+00,  8.4230e+00],\n",
       "           [ 2.3739e+01, -1.9797e+01,  3.2842e+01,  ...,  8.3792e-01,\n",
       "             1.6034e+01, -1.0638e+01],\n",
       "           ...,\n",
       "           [-1.9486e+01,  1.2989e+00,  2.7964e-01,  ...,  1.1599e+01,\n",
       "            -2.0324e+01, -1.1203e+01],\n",
       "           [ 1.3059e+01,  1.9184e+00,  1.7263e+01,  ..., -2.5808e+01,\n",
       "            -4.3004e+01,  1.7798e+01],\n",
       "           [-8.6877e+00,  3.8323e+00, -1.2414e+01,  ..., -5.6531e-01,\n",
       "             1.7289e+01,  1.1021e+01]],\n",
       "\n",
       "          [[ 4.6336e+01, -2.1755e+01,  7.4148e+00,  ..., -1.7586e+01,\n",
       "            -5.8143e-01,  1.9091e+01],\n",
       "           [-1.0220e+00,  8.9537e+00,  3.5386e+01,  ..., -4.0233e+00,\n",
       "            -3.6896e+00, -3.6225e+00],\n",
       "           [-3.4457e+00,  2.5113e+01, -1.5173e+01,  ...,  4.4130e+00,\n",
       "            -1.9792e+01,  1.7315e+01],\n",
       "           ...,\n",
       "           [-7.7569e+00, -8.3307e-01,  1.0658e+01,  ...,  6.1009e+00,\n",
       "            -5.7906e+00,  2.4321e+00],\n",
       "           [-5.2546e+00,  9.7889e-01,  1.0753e+00,  ..., -1.5457e+01,\n",
       "             1.4002e+01, -1.9658e+01],\n",
       "           [-8.5992e+00,  7.8643e+00,  3.2741e+01,  ..., -2.1746e+01,\n",
       "            -1.1742e+01,  9.9090e+00]],\n",
       "\n",
       "          [[ 7.7484e+00,  2.3853e+01, -2.7819e+00,  ...,  1.7260e+01,\n",
       "             1.1802e+01,  1.9042e+01],\n",
       "           [ 1.6759e+01,  1.1538e+01,  3.0025e+01,  ..., -2.0754e+01,\n",
       "             3.5517e+01, -3.8276e+00],\n",
       "           [-3.8604e+01,  2.0427e+01,  1.4532e+01,  ...,  3.4068e+01,\n",
       "            -5.3226e+00, -3.3558e+01],\n",
       "           ...,\n",
       "           [ 3.6082e+00, -4.7401e+00, -9.9772e+00,  ...,  2.6409e+01,\n",
       "             1.0925e+00,  3.7864e+01],\n",
       "           [ 1.3566e+01, -3.5604e+00, -1.2519e+01,  ..., -3.6974e-01,\n",
       "            -1.0363e+01, -2.1081e+01],\n",
       "           [-2.4591e+01, -1.5254e+01, -3.6995e+01,  ..., -1.6433e+01,\n",
       "            -1.9044e+01, -4.2167e+01]]],\n",
       "\n",
       "\n",
       "         [[[-2.0255e+01,  1.4878e+01,  3.5342e+00,  ..., -1.1382e+01,\n",
       "             1.7931e+00, -5.0501e+00],\n",
       "           [ 1.4590e+01,  9.3706e+00, -3.6457e+00,  ...,  1.8701e+00,\n",
       "             2.2034e+01, -3.2279e+01],\n",
       "           [ 1.8244e+01, -1.3069e+01,  3.1326e+01,  ...,  1.1953e+01,\n",
       "             1.8086e+01,  1.1740e+01],\n",
       "           ...,\n",
       "           [ 4.7527e+00,  3.0511e+01, -1.0886e+01,  ...,  2.6109e+01,\n",
       "             2.5973e-01, -8.1333e+00],\n",
       "           [ 1.5906e+01, -3.0952e+01, -2.4916e+01,  ...,  1.2920e+01,\n",
       "            -1.6620e+00,  4.3493e+00],\n",
       "           [ 6.1274e+00,  1.6258e+01,  1.3347e+01,  ..., -3.2710e+00,\n",
       "             1.5465e+01, -2.7261e+01]],\n",
       "\n",
       "          [[-2.6547e+01,  2.6187e+01, -5.3748e+00,  ..., -5.7834e+00,\n",
       "            -1.4342e+01,  1.9449e+01],\n",
       "           [-2.8319e+01, -4.1049e+01, -1.6282e+01,  ...,  7.0555e+00,\n",
       "            -1.7613e+01, -3.8944e+01],\n",
       "           [-2.2652e+01,  1.7243e+01, -1.3611e+01,  ...,  4.8434e+01,\n",
       "            -3.7040e+00,  3.2076e+00],\n",
       "           ...,\n",
       "           [-1.7573e+01,  4.4522e+01, -7.2996e+00,  ...,  1.9016e+01,\n",
       "             8.1669e+00, -1.4878e+01],\n",
       "           [ 2.1276e+01,  1.2754e+01, -7.4132e+00,  ...,  2.7704e+01,\n",
       "            -2.8020e+00, -3.0366e+00],\n",
       "           [ 6.3941e+00, -4.4264e+00, -1.7663e+01,  ..., -4.1858e+00,\n",
       "            -2.8084e+01,  1.5152e+01]],\n",
       "\n",
       "          [[ 3.9581e+00,  2.0032e+01,  1.2723e+01,  ...,  1.5817e+01,\n",
       "            -2.7981e+01, -1.1411e+01],\n",
       "           [ 3.3233e+01, -2.1694e+00,  2.9576e+01,  ..., -2.2580e+01,\n",
       "             3.4122e+01,  1.6128e+01],\n",
       "           [ 8.0671e+00,  2.1571e+01, -1.7664e+01,  ...,  2.4783e+01,\n",
       "             1.5007e+01,  1.8332e+01],\n",
       "           ...,\n",
       "           [-1.0616e+01, -4.2216e+01,  2.6087e+01,  ...,  3.0229e+00,\n",
       "             1.7095e+01,  1.1889e+01],\n",
       "           [-3.7575e+00,  9.4202e+00,  1.0431e+01,  ..., -4.1733e+01,\n",
       "             8.7493e+00,  6.7048e+00],\n",
       "           [ 1.4406e+01,  1.3376e+01,  2.1306e+01,  ...,  7.7736e+00,\n",
       "            -1.2522e+01,  3.4364e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 6.5326e+00,  2.0437e+01,  1.2807e+01,  ..., -3.7699e+01,\n",
       "             3.0154e+01, -2.5483e+01],\n",
       "           [-1.1459e+01,  1.2124e+01,  2.5260e+01,  ..., -2.2068e+01,\n",
       "            -1.6297e+01,  1.0567e+01],\n",
       "           [ 9.6796e+00,  2.5513e+00,  1.0045e+01,  ..., -3.0182e+01,\n",
       "            -5.4332e+00,  3.4221e+01],\n",
       "           ...,\n",
       "           [-6.7110e-03,  3.1461e+01, -5.2368e+00,  ...,  1.1820e+01,\n",
       "            -6.7573e+00, -9.5008e+00],\n",
       "           [ 2.8214e+01, -8.9937e+00,  5.0492e+01,  ..., -8.4329e+00,\n",
       "            -2.1271e+01,  2.6729e+01],\n",
       "           [-2.5303e+01,  5.0899e+00, -1.0837e+01,  ..., -1.9977e+00,\n",
       "            -6.9819e+00, -1.9971e+01]],\n",
       "\n",
       "          [[ 1.3661e+01,  3.0093e+01,  4.8170e+00,  ...,  7.7969e+00,\n",
       "             4.4683e+01,  3.0419e+00],\n",
       "           [-1.2769e+01, -2.3349e+01,  5.9512e+00,  ..., -2.9780e+00,\n",
       "            -1.1030e+01,  1.4029e+01],\n",
       "           [-1.4586e+01, -2.2432e+01,  2.4096e+01,  ..., -1.0467e+01,\n",
       "            -3.9782e+01,  8.5658e+00],\n",
       "           ...,\n",
       "           [-2.2581e+01, -2.0365e+01, -2.7234e+00,  ..., -1.1863e+01,\n",
       "            -9.6514e+00, -2.2752e+01],\n",
       "           [ 8.4708e+00,  1.9001e+00,  1.2605e+01,  ...,  1.6466e+00,\n",
       "             2.5650e+01, -1.2224e+01],\n",
       "           [ 4.1147e+01, -7.2351e+00, -1.0766e+01,  ..., -6.6185e+00,\n",
       "            -1.9495e+01, -1.4265e+01]],\n",
       "\n",
       "          [[ 3.0844e+01,  9.1633e+00, -2.1006e+00,  ...,  2.6238e+01,\n",
       "             4.4535e+00,  1.3355e+01],\n",
       "           [-9.0967e+00, -2.4542e+00, -2.3826e+01,  ..., -1.2148e+01,\n",
       "             1.4459e+01, -3.8855e+01],\n",
       "           [-1.7000e+01, -2.4777e+01,  2.3249e+01,  ...,  6.7272e+00,\n",
       "             1.7293e+01,  4.5214e+01],\n",
       "           ...,\n",
       "           [ 1.0451e+01,  9.7721e+00,  7.9924e+00,  ..., -8.8192e+00,\n",
       "            -7.3724e+00,  1.1365e+01],\n",
       "           [-3.2426e+01, -2.1752e+01,  3.4239e+01,  ..., -2.8930e+00,\n",
       "             1.4103e+01, -2.1813e+01],\n",
       "           [ 5.4724e+00,  1.0244e+01, -1.9187e+01,  ...,  1.9741e+00,\n",
       "            -1.7254e+01,  1.7481e+01]]],\n",
       "\n",
       "\n",
       "         [[[ 9.7446e-01,  1.0070e+01, -2.0387e+01,  ...,  6.0189e+00,\n",
       "             4.0935e+00,  1.2204e+01],\n",
       "           [ 8.5956e-01, -2.9726e+01,  1.6289e+01,  ...,  1.1769e+01,\n",
       "            -1.1048e+01,  1.4488e+01],\n",
       "           [ 2.6103e+01, -5.2116e-01,  1.4104e+01,  ..., -7.1463e+00,\n",
       "             2.0342e+01, -1.2641e+01],\n",
       "           ...,\n",
       "           [-1.7024e+01, -3.1183e+00,  3.0449e+01,  ..., -1.2011e+01,\n",
       "            -3.9064e+01, -3.9251e+01],\n",
       "           [-3.1357e+01, -2.2650e-01, -6.8654e+00,  ..., -3.0015e+01,\n",
       "            -1.0030e+01, -1.8877e+01],\n",
       "           [ 1.5391e+01, -3.2147e+01,  5.8086e+00,  ..., -1.7598e+01,\n",
       "            -1.7528e+01, -2.6720e+01]],\n",
       "\n",
       "          [[-7.6736e+00,  3.6555e+01,  3.6205e+01,  ...,  4.8770e+01,\n",
       "            -1.9958e+01,  4.2053e+01],\n",
       "           [-6.6915e+00,  9.5598e+00, -2.2095e+00,  ...,  2.4570e-01,\n",
       "             1.7953e+01,  2.5246e+01],\n",
       "           [ 1.9728e+01,  1.7187e+00,  6.7670e+01,  ..., -3.4890e+01,\n",
       "            -1.9871e+01,  2.4910e+01],\n",
       "           ...,\n",
       "           [ 4.1891e+01, -2.7621e+01, -1.8723e+01,  ...,  1.5747e+01,\n",
       "            -1.7008e+01, -2.4930e+01],\n",
       "           [-9.7279e+00,  4.7346e+01,  4.7040e+01,  ...,  3.3585e+00,\n",
       "            -3.5007e+01,  2.7268e+01],\n",
       "           [-1.1493e+01, -3.3526e+00, -8.5143e+00,  ...,  2.2907e+01,\n",
       "            -1.9910e+00,  1.0760e+01]],\n",
       "\n",
       "          [[ 2.4137e+01,  5.6174e+00, -2.2734e+00,  ..., -2.9109e+01,\n",
       "            -1.2372e+01, -1.6129e+01],\n",
       "           [-4.2760e+01, -2.1160e+01,  3.3728e+01,  ...,  1.6627e+01,\n",
       "            -2.7195e+01,  2.2482e+01],\n",
       "           [ 4.0330e+01, -1.2719e+01, -8.6080e-01,  ..., -8.0474e+00,\n",
       "            -1.7961e+00, -4.1635e+01],\n",
       "           ...,\n",
       "           [ 1.5179e+01,  1.4479e+01, -5.1248e+00,  ..., -1.5386e+01,\n",
       "            -9.0339e+00,  5.0417e+01],\n",
       "           [-1.3601e+01,  4.8831e+00, -1.1816e+01,  ..., -9.9636e+00,\n",
       "             1.3317e+01, -4.2525e+01],\n",
       "           [-2.3404e+01, -7.9262e+00, -7.2962e-01,  ...,  2.3104e+01,\n",
       "            -1.9007e+01, -1.1179e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.7040e+01, -7.3873e+00,  3.6256e+01,  ...,  1.7579e+01,\n",
       "            -3.6319e+00,  1.9469e+01],\n",
       "           [ 4.2298e+00,  4.0784e+00,  2.0503e+01,  ..., -3.6343e+00,\n",
       "             5.7731e+00, -1.1830e+01],\n",
       "           [ 1.7510e+01, -1.2234e+01, -5.6496e-01,  ..., -3.0995e+00,\n",
       "            -2.9321e+01,  1.9249e+01],\n",
       "           ...,\n",
       "           [-7.2699e-01,  4.0445e+01, -1.5188e+01,  ...,  1.1105e+01,\n",
       "            -1.8381e+01,  2.2990e+01],\n",
       "           [-5.9712e+00, -8.8593e+00,  1.1853e+01,  ...,  7.1178e+00,\n",
       "            -1.5415e+01,  4.7952e+01],\n",
       "           [-1.8074e+01,  1.6615e+01,  2.6743e+01,  ...,  5.1332e+01,\n",
       "             2.6751e+01,  6.8560e+00]],\n",
       "\n",
       "          [[-2.6301e+00, -2.1224e+01,  6.1426e+01,  ..., -1.8547e+01,\n",
       "             2.7911e+01, -3.8329e+01],\n",
       "           [ 3.5869e+00,  2.0316e+01,  1.7159e+01,  ..., -1.1630e+01,\n",
       "            -2.0055e+00,  1.0510e+01],\n",
       "           [ 3.2129e+01, -3.3355e-02,  1.7741e+01,  ..., -1.0551e+01,\n",
       "            -3.2125e+01, -1.3248e+01],\n",
       "           ...,\n",
       "           [ 2.1903e+01, -1.3685e+01, -4.5459e+01,  ...,  8.8053e-01,\n",
       "            -1.1980e+01, -2.9601e+01],\n",
       "           [-1.5089e+01,  3.8323e+01,  2.2777e+01,  ...,  3.5573e+01,\n",
       "             2.7092e+01,  5.8844e+00],\n",
       "           [ 2.5795e+01,  3.0522e+01, -1.7344e+01,  ..., -3.0185e+01,\n",
       "            -1.5985e+01,  6.7970e-01]],\n",
       "\n",
       "          [[-2.3502e+01,  1.3917e+01,  2.7064e+00,  ...,  1.0556e+01,\n",
       "            -2.3155e+01, -8.9421e+00],\n",
       "           [ 4.2844e+00, -1.2733e+01,  1.3197e+01,  ..., -2.5799e+00,\n",
       "             1.3626e+01,  3.3363e+01],\n",
       "           [-3.2434e+00,  1.5865e+01,  1.3317e+01,  ..., -2.6229e+01,\n",
       "             2.7803e+01,  4.5029e+00],\n",
       "           ...,\n",
       "           [-1.4818e+01,  4.0649e+01, -8.6568e+00,  ..., -1.0393e+01,\n",
       "            -3.0672e+01,  5.0708e+01],\n",
       "           [-6.8387e+00, -2.3852e+01, -8.3093e+00,  ..., -9.9056e+00,\n",
       "             1.4273e+01,  2.5865e+00],\n",
       "           [ 1.5161e+01, -1.7341e+01,  4.5682e-01,  ...,  1.6865e+01,\n",
       "            -7.8347e+00,  1.8137e+01]]]],\n",
       "\n",
       "\n",
       "\n",
       "        [[[[ 1.5557e+01,  3.0575e-01, -1.6280e+01,  ...,  6.0422e+01,\n",
       "             5.5571e+00,  2.6097e+01],\n",
       "           [-3.3076e+01,  1.7843e+01,  2.2286e+01,  ...,  3.6660e+00,\n",
       "            -7.9512e+00,  1.8872e+01],\n",
       "           [ 5.6864e+00,  1.1481e+01,  1.0663e+00,  ...,  1.4418e+01,\n",
       "             3.3613e+01, -3.8907e+01],\n",
       "           ...,\n",
       "           [-3.8333e+01, -9.9290e+00,  1.2832e+01,  ...,  1.5743e+01,\n",
       "            -6.5092e+00,  3.2312e+00],\n",
       "           [-1.4923e+01,  1.3266e+01,  2.2047e+01,  ...,  2.4948e+01,\n",
       "             6.6777e+00,  1.1801e+01],\n",
       "           [-3.0653e+01,  6.3585e+00,  3.4177e+01,  ..., -3.2098e+00,\n",
       "             3.6625e+01, -2.5395e+01]],\n",
       "\n",
       "          [[-2.3341e-01, -6.6914e+00,  6.9049e-01,  ...,  2.2579e+00,\n",
       "            -9.3475e+00,  8.1392e-01],\n",
       "           [-1.8820e+01, -5.8802e+00,  2.2450e+01,  ..., -1.1048e+01,\n",
       "            -7.6094e+00, -3.1022e+01],\n",
       "           [-3.0976e+01,  1.3513e+01,  2.1012e+01,  ..., -2.1730e+01,\n",
       "             5.7619e+00,  1.8455e+01],\n",
       "           ...,\n",
       "           [-1.1586e+01, -2.3324e+01, -2.4374e+01,  ..., -1.5014e+01,\n",
       "            -1.9330e+01, -8.0927e+00],\n",
       "           [-1.6588e+01,  2.3182e+00,  1.2479e+01,  ...,  1.7704e+01,\n",
       "            -3.0039e+01, -1.6945e+00],\n",
       "           [-1.9585e+01,  6.3700e+00, -5.9728e+00,  ..., -9.0438e+00,\n",
       "            -1.4392e+01,  3.1227e+01]],\n",
       "\n",
       "          [[-2.2648e+01, -8.0979e+00, -4.1832e+01,  ...,  1.6351e+01,\n",
       "             2.0141e+01,  7.8587e+00],\n",
       "           [-1.8109e+01, -1.8380e+01,  6.7005e+00,  ...,  9.2661e+00,\n",
       "            -3.3548e+00,  1.0049e+01],\n",
       "           [-2.8688e+00,  2.2969e+01,  1.2347e+01,  ...,  3.1636e+01,\n",
       "            -1.9164e+01, -5.4090e+00],\n",
       "           ...,\n",
       "           [-9.0506e+00,  1.0366e+00, -3.9521e+00,  ..., -1.6194e+01,\n",
       "             2.9029e+01,  3.4210e+01],\n",
       "           [-1.6710e+01, -9.9815e+00, -4.3242e+01,  ...,  1.2574e+01,\n",
       "             5.0134e+00, -1.8390e+01],\n",
       "           [ 2.7271e+00,  1.9242e+01, -5.9010e-01,  ..., -1.1395e+01,\n",
       "            -2.8606e+01, -2.2432e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-7.7795e+00, -1.0293e+01,  1.5332e+01,  ...,  3.0521e+01,\n",
       "             8.0587e+00, -3.1092e+00],\n",
       "           [-2.4385e+01,  3.4224e+01, -1.3884e+01,  ..., -3.0997e+01,\n",
       "            -2.1774e+01,  1.9234e+01],\n",
       "           [ 2.2926e+01,  9.6331e+00, -4.1443e+00,  ..., -6.4604e-01,\n",
       "            -1.7260e+01,  1.3798e+01],\n",
       "           ...,\n",
       "           [-1.0864e+01, -1.2002e+00,  7.2344e+00,  ...,  1.8724e+01,\n",
       "             1.4170e+01,  4.4162e+01],\n",
       "           [ 3.4006e+01, -1.6582e+01,  3.8535e+01,  ...,  2.9547e+00,\n",
       "             4.0092e+01, -8.7703e+00],\n",
       "           [ 1.7319e+01, -4.5467e+00, -8.9517e+00,  ..., -1.4250e+01,\n",
       "            -5.3235e+00, -4.3914e+01]],\n",
       "\n",
       "          [[-3.7637e+00,  5.9589e+01, -1.3638e+01,  ...,  6.9533e+00,\n",
       "             2.5139e+01,  4.2182e+01],\n",
       "           [-1.8901e+01, -4.5430e+00, -6.6494e+00,  ...,  1.3080e+01,\n",
       "             1.9294e+01,  2.0749e+01],\n",
       "           [ 2.7018e+00,  1.4431e+01, -1.6803e+01,  ..., -2.0446e+01,\n",
       "            -1.6181e+01,  1.7026e+01],\n",
       "           ...,\n",
       "           [-7.3630e+00,  7.6670e+01, -1.1339e+01,  ..., -2.2077e+01,\n",
       "            -2.2854e+01,  1.0850e+01],\n",
       "           [-1.1385e+01,  1.4982e+01,  3.0986e+01,  ..., -1.3846e+01,\n",
       "             6.8928e+00,  1.4323e+01],\n",
       "           [-2.7347e+00, -3.1933e-01, -1.5680e+01,  ..., -8.6589e+00,\n",
       "             2.3257e+00,  1.6159e+01]],\n",
       "\n",
       "          [[ 2.6204e+01,  2.8616e+01, -4.2591e-01,  ..., -3.6087e+01,\n",
       "             3.9414e+00,  1.2480e+01],\n",
       "           [-9.4500e+00, -9.9551e+00,  1.8056e+01,  ...,  1.8427e+01,\n",
       "            -1.2344e+01,  2.7734e+01],\n",
       "           [-8.4335e+00,  7.0687e+00, -2.2738e+01,  ..., -2.8970e+01,\n",
       "            -5.5775e+00, -1.9467e+01],\n",
       "           ...,\n",
       "           [ 4.7546e+01, -6.5589e+00,  3.0595e+01,  ...,  1.6187e+01,\n",
       "            -9.4057e+00,  1.5378e+01],\n",
       "           [ 1.9902e+00,  2.9266e+01,  1.7675e+01,  ..., -3.0727e+01,\n",
       "            -3.7134e+00, -1.8902e+01],\n",
       "           [-6.2921e+01, -3.8457e+01,  5.1509e+01,  ...,  1.9301e+01,\n",
       "             2.3723e-01, -3.5385e+00]]],\n",
       "\n",
       "\n",
       "         [[[ 2.5117e+01,  6.4039e-02,  2.4215e+01,  ..., -9.9509e+00,\n",
       "             2.4456e+01, -1.5351e+01],\n",
       "           [ 7.1758e+00,  4.1922e+00, -8.9076e+00,  ...,  5.9944e+00,\n",
       "             1.6730e+01,  5.6422e+01],\n",
       "           [-5.2748e+00,  3.3513e+01, -5.0521e+01,  ...,  3.0762e+01,\n",
       "            -1.8311e+01, -1.8007e+01],\n",
       "           ...,\n",
       "           [ 2.6060e+01,  3.1455e+01, -3.0438e+01,  ..., -6.4089e+00,\n",
       "             3.3687e+01,  6.4998e+00],\n",
       "           [ 4.2882e+01, -7.4126e+00, -1.1096e+01,  ...,  3.6002e+01,\n",
       "            -2.2252e+00, -6.2991e+00],\n",
       "           [-1.6049e+01, -4.9896e+00,  2.3074e+01,  ...,  3.2224e+00,\n",
       "            -1.5881e+01, -8.7753e+00]],\n",
       "\n",
       "          [[ 1.4650e+01, -1.7565e+01,  1.6303e+01,  ...,  1.3289e+01,\n",
       "             2.0131e+01,  1.8584e+00],\n",
       "           [ 1.5584e+01, -2.8504e+01,  2.3036e+01,  ..., -5.1104e+01,\n",
       "             9.0121e-01, -1.9877e+00],\n",
       "           [ 2.5370e+01,  2.3587e+01, -2.5247e+01,  ...,  1.2643e+01,\n",
       "            -8.5659e+00,  1.1815e+01],\n",
       "           ...,\n",
       "           [-1.3119e+01, -9.0579e+00, -2.9459e+01,  ...,  2.2268e+00,\n",
       "             2.2954e+01, -5.1381e+00],\n",
       "           [ 2.1235e+01,  1.7792e+01,  3.9063e+01,  ..., -1.2174e+01,\n",
       "            -1.6454e+01, -4.6213e+00],\n",
       "           [-4.8960e+00, -1.6980e+01, -2.0370e+01,  ..., -5.4472e-01,\n",
       "            -3.4491e+00,  1.2552e+01]],\n",
       "\n",
       "          [[ 2.9410e+01,  2.2415e+01,  4.3981e+01,  ..., -2.0974e+01,\n",
       "             3.4199e+01, -1.2183e+01],\n",
       "           [-4.6634e+00, -3.7953e+01,  2.5576e+01,  ...,  2.4928e+01,\n",
       "            -1.0901e+01,  1.6989e+01],\n",
       "           [ 1.5194e+01,  1.5118e+01,  5.1219e-01,  ..., -4.4450e+00,\n",
       "            -2.3771e+01,  4.6367e+00],\n",
       "           ...,\n",
       "           [ 2.4223e+01, -5.3875e+00, -2.3729e+00,  ...,  2.6483e+01,\n",
       "            -7.5440e+00,  2.5384e+01],\n",
       "           [-1.0590e+01, -1.0772e+01, -2.5492e+01,  ...,  3.3026e+00,\n",
       "            -1.0821e+01, -1.4455e+00],\n",
       "           [ 5.9761e+00,  1.5574e+01, -1.2692e+01,  ...,  3.4276e+01,\n",
       "             4.4517e+00, -1.5163e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-3.6119e+01, -1.6253e+01, -1.3506e+01,  ..., -9.0221e+00,\n",
       "             1.6461e+01, -1.2129e+01],\n",
       "           [ 1.0609e+01, -5.1574e+00, -8.9131e+00,  ..., -1.6921e+01,\n",
       "             1.1447e+01, -2.6456e+01],\n",
       "           [ 4.8765e+00, -3.1378e+01, -2.7175e+01,  ..., -3.7469e+01,\n",
       "             7.1332e+00,  2.7865e+01],\n",
       "           ...,\n",
       "           [ 2.9663e+01, -1.6221e+01, -5.9436e+00,  ...,  1.0859e+01,\n",
       "             1.6159e+01,  6.2639e-01],\n",
       "           [-3.5930e+01, -1.9588e+01,  4.2516e+00,  ..., -5.7757e+00,\n",
       "            -1.2162e+01, -1.0439e+01],\n",
       "           [ 2.4379e+01,  2.4089e+01,  1.6195e+01,  ..., -2.8948e+00,\n",
       "             1.9109e+00,  1.2395e+01]],\n",
       "\n",
       "          [[ 2.9365e+01, -1.7021e+01, -4.1669e+01,  ...,  4.3286e+00,\n",
       "            -2.5090e+01, -5.9229e+01],\n",
       "           [ 1.3010e+01,  1.3361e+01,  1.2726e+01,  ...,  2.8218e+01,\n",
       "             2.1120e+01, -2.3133e+01],\n",
       "           [-2.5260e+01, -2.2353e+01, -2.2138e+01,  ...,  2.4057e+01,\n",
       "             1.5525e+01, -1.8355e+01],\n",
       "           ...,\n",
       "           [ 5.2051e+01, -1.4292e+01, -3.8579e+01,  ..., -2.2503e+01,\n",
       "            -6.7339e+00,  6.6680e+00],\n",
       "           [-3.3560e+00,  3.4497e+01,  9.9002e-01,  ...,  1.5491e+01,\n",
       "             8.0967e+00, -1.4836e+01],\n",
       "           [-9.6653e+00, -1.6411e+01, -7.3704e+00,  ...,  6.3138e+00,\n",
       "            -3.7512e+00, -7.4437e+00]],\n",
       "\n",
       "          [[-2.4063e+00,  5.6985e+00,  1.9492e+01,  ...,  3.4186e+00,\n",
       "             3.2671e+01, -2.4585e+01],\n",
       "           [-2.2472e+01, -7.8109e+00,  1.9161e+01,  ...,  2.8717e+01,\n",
       "             2.2822e+00, -1.1282e+01],\n",
       "           [ 1.1601e+01, -1.0601e+01,  5.5170e+00,  ..., -1.9169e+00,\n",
       "             4.7793e+00,  2.5845e+01],\n",
       "           ...,\n",
       "           [-4.3413e+01, -1.3150e+01,  1.5623e+01,  ..., -1.1904e-01,\n",
       "             9.5689e+00,  3.8862e+00],\n",
       "           [-1.0258e+01,  2.9097e+01, -6.1895e+00,  ..., -3.1495e+01,\n",
       "            -6.0550e+00,  1.6893e+01],\n",
       "           [ 1.9353e+00,  4.0113e+01,  2.4711e+01,  ...,  2.5706e+01,\n",
       "            -2.0550e+01, -1.7774e+01]]],\n",
       "\n",
       "\n",
       "         [[[-9.0599e+00,  3.2127e+01, -2.8538e+00,  ...,  4.9598e+00,\n",
       "            -7.2272e-01, -1.6226e+01],\n",
       "           [ 4.1429e+00,  9.1090e+00, -3.7382e+01,  ..., -2.4871e+01,\n",
       "            -5.5787e+01, -2.2146e+01],\n",
       "           [ 6.6319e+00,  1.3638e+01,  3.5704e+01,  ..., -2.4609e+00,\n",
       "             1.1438e+01,  3.3504e+01],\n",
       "           ...,\n",
       "           [-5.5292e+00,  8.6872e+00,  2.4783e+01,  ..., -3.4307e+01,\n",
       "            -1.0829e+01,  1.6921e+01],\n",
       "           [-3.1931e+01,  1.6933e+01, -5.5835e+01,  ...,  1.7104e+01,\n",
       "             3.5168e+01,  1.8229e+01],\n",
       "           [ 2.0231e+01, -1.5186e+01,  4.3364e+01,  ...,  1.8661e+01,\n",
       "            -2.8677e-03,  9.7212e+00]],\n",
       "\n",
       "          [[ 2.2870e+01,  1.9108e+01,  1.0228e+01,  ..., -2.8191e+01,\n",
       "            -2.4414e+01,  1.2655e+01],\n",
       "           [-1.3526e+01, -1.2431e+01,  3.0083e+00,  ..., -3.5667e+01,\n",
       "            -8.9658e+00,  8.4847e+00],\n",
       "           [ 1.8327e+00, -2.9330e+01,  6.5518e+00,  ...,  2.5015e+01,\n",
       "            -2.1597e+01,  1.0082e+01],\n",
       "           ...,\n",
       "           [ 1.4233e+01, -2.0658e-02, -2.8203e+01,  ..., -4.9068e+01,\n",
       "             1.3672e+01,  1.4619e+01],\n",
       "           [ 2.2339e+01,  1.1460e+01, -3.5651e+01,  ...,  3.8608e+00,\n",
       "            -2.8375e+01,  7.5938e+00],\n",
       "           [ 1.4941e+01,  9.7592e+00,  1.3317e+01,  ..., -2.0264e+01,\n",
       "            -2.3758e+00,  2.4279e+01]],\n",
       "\n",
       "          [[ 2.2877e+01,  7.5932e-01,  4.3925e+01,  ..., -3.6141e+01,\n",
       "            -1.9819e+01, -6.2961e+00],\n",
       "           [-1.8737e+01, -6.3872e+00, -1.2724e+00,  ...,  3.6121e+00,\n",
       "             2.2381e+00,  2.7711e+00],\n",
       "           [-7.3968e+00,  2.3613e+01,  8.8195e+00,  ...,  3.5343e+01,\n",
       "            -1.2907e+01,  2.9205e+01],\n",
       "           ...,\n",
       "           [-4.9628e+01,  9.7869e+00,  1.7640e+01,  ..., -4.2943e+01,\n",
       "             1.9712e+01,  7.1568e+01],\n",
       "           [ 3.7959e+01, -5.4299e+01,  4.0084e+01,  ..., -3.6901e+01,\n",
       "             1.8429e+01, -3.3149e+01],\n",
       "           [ 2.5917e+01,  2.1715e+01,  3.7997e+01,  ..., -2.4785e+01,\n",
       "            -2.1959e+01,  8.1025e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-9.9306e+00, -3.8423e+01,  1.6433e+01,  ..., -3.3183e+00,\n",
       "             7.4106e+00,  1.7060e+01],\n",
       "           [-1.5166e+01,  6.5134e+00, -3.0876e+00,  ...,  1.4683e+01,\n",
       "            -2.2810e+01,  2.7327e+01],\n",
       "           [-1.3511e+01,  7.7539e-02,  1.8550e+01,  ...,  3.0955e+01,\n",
       "            -1.3324e+01,  3.5464e+01],\n",
       "           ...,\n",
       "           [-7.8052e+00,  1.5591e+01, -6.2426e+00,  ..., -3.5694e+01,\n",
       "             8.3407e+00,  5.3765e+00],\n",
       "           [-3.0030e+01,  1.6988e+01,  5.2443e+00,  ..., -4.6850e+00,\n",
       "            -2.4632e+01,  3.7723e+01],\n",
       "           [-1.1177e+00,  2.8454e+01, -4.6739e+01,  ...,  3.1995e+01,\n",
       "             3.6639e+01,  6.6863e+00]],\n",
       "\n",
       "          [[-2.2047e+01,  8.1395e-02,  1.8759e+00,  ...,  3.2878e-01,\n",
       "            -3.3022e+01,  3.5069e+01],\n",
       "           [ 2.0511e+01, -2.7794e+01, -1.5128e+01,  ..., -2.7550e+00,\n",
       "             3.2938e+01, -7.6706e+00],\n",
       "           [-3.0994e+01, -3.1989e+01,  2.2132e+00,  ..., -3.8974e+01,\n",
       "            -2.4003e+01, -4.7080e+00],\n",
       "           ...,\n",
       "           [-2.6484e+01,  4.2945e+00, -1.7080e+01,  ...,  1.1124e+01,\n",
       "            -2.5857e+01, -2.3816e+01],\n",
       "           [ 3.4306e+00, -2.0759e+00,  3.3717e+01,  ..., -4.4844e+00,\n",
       "            -1.7311e+01, -4.8927e+00],\n",
       "           [ 1.5805e+01, -2.8689e+01, -2.7465e+01,  ...,  2.4667e+01,\n",
       "             3.9132e-01,  1.0696e+01]],\n",
       "\n",
       "          [[ 2.6511e+01, -2.2387e+01, -1.8321e+01,  ...,  1.7570e+01,\n",
       "            -1.3282e+01,  4.5107e+01],\n",
       "           [ 1.0102e+01, -3.0799e+01, -1.1255e+01,  ...,  1.3104e+01,\n",
       "             3.8965e+01, -4.6917e+01],\n",
       "           [-2.8403e+00, -7.5107e+00, -3.9556e+00,  ...,  3.3564e+01,\n",
       "            -3.1224e+01,  4.5418e+01],\n",
       "           ...,\n",
       "           [-7.7910e+00,  4.2450e+00, -4.0861e+00,  ...,  3.0471e+01,\n",
       "            -2.1716e+01, -7.1102e+00],\n",
       "           [-1.1626e+01,  9.9420e+00, -1.4739e+01,  ..., -2.6105e+01,\n",
       "            -1.7228e+01, -3.4035e+01],\n",
       "           [-3.7484e+01,  9.1316e+00, -2.6238e+01,  ..., -4.6701e+00,\n",
       "             2.5016e+00, -2.4676e+01]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[-1.4808e+01, -3.7263e+01,  8.7582e+00,  ..., -1.9018e+01,\n",
       "             1.0267e+01,  1.3008e+01],\n",
       "           [ 9.0435e+00, -1.6355e+01,  3.8078e+00,  ..., -5.2675e+00,\n",
       "            -1.5605e+01, -9.2638e+00],\n",
       "           [ 2.1621e+01, -3.5554e+01,  3.6336e-01,  ...,  9.7230e+00,\n",
       "            -1.2439e+01,  1.1575e+01],\n",
       "           ...,\n",
       "           [-3.8099e+01,  1.7770e+01, -5.5715e+00,  ..., -1.0092e+01,\n",
       "             3.7895e+00, -3.7511e+01],\n",
       "           [ 5.8927e+00, -1.2107e+01,  1.1835e+01,  ...,  5.5006e+00,\n",
       "            -4.8574e+01, -9.1753e+00],\n",
       "           [-1.2843e+01, -1.3408e+01,  1.6496e+01,  ..., -1.6769e+00,\n",
       "            -8.7728e+00, -2.2301e+01]],\n",
       "\n",
       "          [[ 4.0212e+01,  1.1632e+01, -1.4623e+01,  ...,  1.9886e+01,\n",
       "            -1.4768e+01, -3.5357e+01],\n",
       "           [ 1.2367e+01, -1.3578e+01, -2.9950e+00,  ...,  1.4217e+00,\n",
       "            -3.0542e+01, -5.1959e+00],\n",
       "           [-1.1658e+01,  1.4633e+00,  1.2235e+01,  ..., -5.2081e+00,\n",
       "            -4.9221e+01,  9.3921e+00],\n",
       "           ...,\n",
       "           [ 5.4994e+00, -1.1528e+01, -1.5566e+01,  ...,  2.5508e+01,\n",
       "            -1.6944e-01, -1.0439e+01],\n",
       "           [-2.9956e+01,  8.4381e+00,  8.6419e+00,  ..., -3.3057e+00,\n",
       "             1.1258e+01, -4.5377e+01],\n",
       "           [ 1.8337e+01,  1.2132e+01,  4.8841e+00,  ...,  1.0442e+01,\n",
       "             1.0837e+01, -4.2890e+01]],\n",
       "\n",
       "          [[-3.9085e+01, -1.3636e+01,  2.1838e+01,  ...,  4.8253e+01,\n",
       "            -2.3663e+00, -8.4176e+00],\n",
       "           [-2.0860e+01, -8.3220e+00,  9.1171e+00,  ...,  7.3274e+00,\n",
       "            -1.1957e+01, -2.6164e+01],\n",
       "           [-2.9453e+01, -2.3510e+01,  2.4187e+01,  ..., -2.9335e+01,\n",
       "             3.3765e+01, -8.1338e+00],\n",
       "           ...,\n",
       "           [ 7.2995e+00,  2.3652e+01,  6.0532e+00,  ..., -1.1832e+01,\n",
       "             3.8929e+00, -2.1526e+01],\n",
       "           [-2.1186e+01,  3.6750e+00,  4.2137e+01,  ...,  1.5748e+01,\n",
       "            -8.3594e+00, -2.4963e+01],\n",
       "           [ 4.3186e+00, -8.5452e+00, -6.2786e+00,  ..., -6.7524e+00,\n",
       "            -4.2478e+00,  2.4498e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.8932e+01, -3.3708e+01, -5.3692e+00,  ...,  1.1210e+01,\n",
       "             1.2032e+01,  5.5757e-01],\n",
       "           [-2.2658e+01,  1.8856e+01, -5.4889e+01,  ...,  1.4032e+01,\n",
       "            -6.7617e+00,  5.7853e+00],\n",
       "           [ 9.1865e-01, -2.5562e+01,  3.0984e+01,  ..., -8.4019e+00,\n",
       "            -5.7736e+00, -1.5450e+01],\n",
       "           ...,\n",
       "           [-2.4082e+01, -9.7314e+00,  3.2908e+00,  ..., -2.0597e+01,\n",
       "            -2.2638e+01,  2.6201e+00],\n",
       "           [ 8.0569e+00,  1.4585e+01,  1.5121e+01,  ..., -1.0336e+01,\n",
       "             9.4268e+00,  3.8329e+01],\n",
       "           [ 8.7051e+00,  3.5347e+00, -4.5541e+01,  ...,  1.6875e+01,\n",
       "             1.7704e+01,  9.9360e+00]],\n",
       "\n",
       "          [[ 1.3108e+00,  4.0693e+00,  4.0429e+01,  ...,  4.8246e+01,\n",
       "            -1.1544e+01, -3.9107e+00],\n",
       "           [ 6.5670e+00,  8.2032e+00,  8.5856e-01,  ...,  3.8033e+01,\n",
       "            -5.5802e+00,  3.6166e+00],\n",
       "           [ 6.4106e+00, -1.7424e+01, -3.1307e-01,  ..., -1.9648e+00,\n",
       "             2.6457e+01, -1.0788e+01],\n",
       "           ...,\n",
       "           [ 2.4767e+00, -1.1643e+00, -2.6780e+01,  ...,  5.7198e-01,\n",
       "             3.3493e+01,  1.3499e+01],\n",
       "           [ 2.9077e+01, -1.2034e+01,  2.3005e+01,  ...,  2.5415e+00,\n",
       "             2.4492e+01,  2.1273e+01],\n",
       "           [-1.2553e+01, -2.4520e+01,  2.7363e+01,  ...,  3.2627e+01,\n",
       "             3.5074e+00, -3.8349e+01]],\n",
       "\n",
       "          [[-1.3460e+01, -5.5838e+00, -1.9303e+01,  ...,  5.0816e+01,\n",
       "             1.5831e+01, -9.4064e+00],\n",
       "           [ 1.4706e+01, -1.8901e+01,  3.8495e+00,  ..., -5.1450e+00,\n",
       "            -1.7550e+01, -2.5070e+01],\n",
       "           [-1.7637e+01, -7.8779e+00,  3.3914e+00,  ..., -1.4679e+01,\n",
       "            -2.1839e+01,  1.3265e+01],\n",
       "           ...,\n",
       "           [-4.0337e+01,  1.4523e+01, -1.2125e+01,  ..., -3.5336e+01,\n",
       "             3.8107e+01, -1.0247e+01],\n",
       "           [-2.3807e+01,  2.5013e+01, -1.7329e+00,  ..., -1.7924e+01,\n",
       "             6.8397e+00, -1.6437e+01],\n",
       "           [ 4.4877e+00, -7.7462e+00,  9.1208e+00,  ...,  1.8117e+01,\n",
       "             1.1051e+01,  2.0675e+01]]],\n",
       "\n",
       "\n",
       "         [[[-6.2477e+00, -2.9078e+01, -3.4082e+01,  ...,  2.2946e+01,\n",
       "            -2.4037e+01,  2.7966e+01],\n",
       "           [-1.5906e+01, -8.1830e+00,  4.6796e+00,  ...,  6.2705e-01,\n",
       "            -1.7801e+01,  2.2346e+01],\n",
       "           [-3.6423e+01,  2.6067e+01,  2.9293e+01,  ..., -1.6503e+01,\n",
       "            -4.6908e+00,  5.0301e+00],\n",
       "           ...,\n",
       "           [ 7.7308e+00,  1.7868e+01, -1.3838e+00,  ..., -3.6191e+00,\n",
       "             1.3699e+01, -1.0023e+01],\n",
       "           [ 2.4294e+01, -2.9643e+00, -1.4537e+01,  ...,  9.5331e+00,\n",
       "            -2.7097e+01,  1.9212e+01],\n",
       "           [ 4.3500e+00, -2.5105e+01,  1.0697e+01,  ..., -1.9685e-01,\n",
       "             2.3165e+01,  8.5474e+00]],\n",
       "\n",
       "          [[-2.8475e+01,  3.1627e+01,  1.2330e+01,  ...,  8.0756e+00,\n",
       "             1.6865e+01,  3.6958e+01],\n",
       "           [ 2.5870e+00, -1.7004e+01,  9.0738e+00,  ...,  1.4802e+01,\n",
       "             1.0662e+01, -1.1290e+01],\n",
       "           [-1.1162e+01, -1.6398e+01, -1.1567e+01,  ..., -3.3186e+00,\n",
       "            -3.8467e+01,  3.8015e+01],\n",
       "           ...,\n",
       "           [-4.8755e+01,  1.0384e+01,  5.8240e+01,  ...,  8.0801e+00,\n",
       "            -6.8498e+00, -1.6100e+01],\n",
       "           [-1.2942e+01,  1.4173e+01, -2.2395e+01,  ..., -1.1442e+01,\n",
       "             1.5761e+01, -5.1703e+00],\n",
       "           [ 8.0650e+00, -2.0980e+01,  1.3267e+01,  ..., -2.2307e+01,\n",
       "            -2.8834e+01, -2.1238e+01]],\n",
       "\n",
       "          [[ 4.3857e+01,  1.0885e+01,  3.6363e+00,  ..., -2.7816e+01,\n",
       "             2.6288e+01, -9.2584e+00],\n",
       "           [ 2.7983e+01,  2.4373e+01,  6.2665e+00,  ..., -2.2138e+01,\n",
       "            -1.0517e+01, -2.1224e+01],\n",
       "           [ 3.7256e+01,  7.4671e+00,  3.0600e+00,  ...,  1.9841e+01,\n",
       "            -3.4369e+01,  2.8729e+01],\n",
       "           ...,\n",
       "           [ 1.8003e+01, -1.0465e+01,  1.7209e+01,  ...,  2.2002e+01,\n",
       "            -2.3463e+01,  4.5870e+01],\n",
       "           [-1.1266e+01,  4.7374e+00, -2.1584e+01,  ..., -4.2945e+01,\n",
       "            -6.5590e-01, -3.0453e+01],\n",
       "           [ 2.5691e+01, -3.7528e+00,  7.1183e+00,  ..., -1.0691e+01,\n",
       "             9.6772e+00,  2.0029e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-1.4068e+01,  7.5627e+00,  2.2845e+01,  ..., -2.8505e+01,\n",
       "             2.6471e+01,  1.1729e+01],\n",
       "           [-1.9322e+01,  2.6278e+01,  3.5120e+01,  ...,  2.1726e+01,\n",
       "             3.8186e+01, -9.7692e+00],\n",
       "           [ 1.9332e+01,  3.1973e+01, -8.9796e+00,  ..., -3.0798e+01,\n",
       "             3.6031e+00,  1.2064e+01],\n",
       "           ...,\n",
       "           [ 7.9418e+00, -1.0032e+01, -7.5095e+00,  ..., -1.3588e+01,\n",
       "            -3.0501e+00,  4.1861e+01],\n",
       "           [-1.4362e+01, -2.5159e+00, -2.1610e+00,  ...,  1.9040e+01,\n",
       "            -2.6489e+01,  5.9014e+00],\n",
       "           [-2.6505e+00,  2.2057e+01, -1.1944e+01,  ...,  6.3522e+00,\n",
       "            -2.4325e+00,  2.1200e+01]],\n",
       "\n",
       "          [[-2.2269e+01, -2.7312e+01,  2.1896e+01,  ..., -8.5793e-01,\n",
       "             6.7148e-01,  1.5054e+00],\n",
       "           [-2.6623e+01,  8.2049e+00,  8.3868e+00,  ..., -1.5035e+01,\n",
       "             1.1702e+01, -1.0608e+01],\n",
       "           [-1.6397e+01, -2.1608e+01, -4.5533e+00,  ..., -8.8944e+00,\n",
       "             1.3070e+01, -2.6182e+01],\n",
       "           ...,\n",
       "           [ 8.3687e+00,  1.7281e+01, -3.9634e+01,  ..., -9.2311e+00,\n",
       "             5.5741e+00, -1.1987e+01],\n",
       "           [-4.9930e+01, -8.4565e+00,  2.3565e+01,  ...,  3.5091e+01,\n",
       "            -1.3722e+01,  1.8099e+01],\n",
       "           [-3.0758e+01,  1.6481e+01, -3.0184e+01,  ...,  2.2909e+01,\n",
       "             9.2560e+00,  9.2971e+00]],\n",
       "\n",
       "          [[ 1.0869e+01, -2.9201e+00, -1.7014e+01,  ...,  1.0991e+00,\n",
       "            -1.8194e+01, -6.4446e-01],\n",
       "           [-1.0358e+01, -1.4847e+01,  1.4877e+01,  ...,  3.5693e+01,\n",
       "             1.5214e+00,  1.3816e+01],\n",
       "           [ 4.1443e+01,  4.3344e+01,  1.8239e+00,  ..., -2.2044e+01,\n",
       "             2.8274e+01, -2.1764e+00],\n",
       "           ...,\n",
       "           [ 2.1115e+00,  1.1445e+01, -1.2635e+01,  ...,  1.2456e+01,\n",
       "            -2.6148e+00, -1.7377e+01],\n",
       "           [ 4.8649e+01,  1.1822e+01,  1.8914e+01,  ..., -1.8970e+01,\n",
       "            -2.4606e+00,  4.5337e+00],\n",
       "           [-7.8710e+00,  2.7211e+00,  1.9219e+01,  ...,  2.2726e+00,\n",
       "             3.0778e+01,  2.9712e+00]]],\n",
       "\n",
       "\n",
       "         [[[-3.8554e+01,  7.9841e+00,  2.5697e+01,  ...,  1.8210e+01,\n",
       "            -2.9656e+01,  6.6701e+00],\n",
       "           [-1.0962e+01,  2.8041e+01, -1.5732e+01,  ..., -1.8864e+01,\n",
       "             2.1034e+00, -7.6743e+00],\n",
       "           [ 1.5876e+01,  2.9756e+01,  4.9406e+00,  ..., -1.5600e+01,\n",
       "             2.9008e+01, -1.2627e+01],\n",
       "           ...,\n",
       "           [ 3.1216e+01, -3.6140e+01,  2.6124e+01,  ..., -1.7970e+01,\n",
       "             2.6977e+00,  1.6738e+01],\n",
       "           [ 2.0413e+01,  1.4969e+01, -3.5775e+00,  ...,  1.2492e+01,\n",
       "            -1.9766e+01, -1.3985e+01],\n",
       "           [ 1.0196e+00, -2.2566e+01, -1.7542e+01,  ...,  4.2113e+01,\n",
       "             2.6537e+01, -2.4192e+01]],\n",
       "\n",
       "          [[-1.0013e+01, -2.3142e+01,  2.7838e+01,  ...,  2.0217e+01,\n",
       "             1.2936e+01, -7.2631e+00],\n",
       "           [-2.9889e+01, -1.3092e+01, -3.0158e+01,  ..., -2.4972e+01,\n",
       "             7.4347e+00, -3.9090e+01],\n",
       "           [ 3.2917e+01,  3.7564e+00, -1.8875e+01,  ..., -1.1588e+01,\n",
       "             2.0485e+01, -2.9717e+00],\n",
       "           ...,\n",
       "           [ 1.6573e+00,  1.3603e+01, -2.2999e+01,  ..., -4.4387e+01,\n",
       "             7.1137e+00,  1.9779e+01],\n",
       "           [ 1.3908e+01,  6.3249e-01, -4.8181e+01,  ..., -9.3662e+00,\n",
       "             3.4933e+00, -1.3377e+01],\n",
       "           [-2.6656e+01,  4.6112e+01, -1.6017e+01,  ..., -1.7045e+01,\n",
       "             9.9375e+00, -5.7933e+01]],\n",
       "\n",
       "          [[-2.7055e+01, -2.9581e+01,  2.6316e+01,  ...,  2.8116e+01,\n",
       "            -2.0530e+01, -7.0607e+00],\n",
       "           [-1.2335e+01,  1.0310e+01, -1.0835e+01,  ..., -3.9650e+01,\n",
       "            -2.0482e+01, -1.9204e+01],\n",
       "           [ 1.3032e+01, -1.7617e+00,  1.7381e+01,  ..., -6.6093e+00,\n",
       "            -1.9404e+01, -1.3322e+01],\n",
       "           ...,\n",
       "           [-2.2661e+01,  2.2685e+01, -3.1678e+01,  ..., -3.1702e+01,\n",
       "             1.8529e+01,  9.3163e+00],\n",
       "           [ 1.5944e+01, -8.9145e+00,  7.7884e+00,  ...,  1.1559e+01,\n",
       "            -2.4230e+01, -2.5397e+01],\n",
       "           [ 1.6506e-01,  1.5469e+01,  3.5387e+00,  ..., -8.7571e+00,\n",
       "             2.0265e+01, -2.5784e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 8.4099e+00, -1.3995e+01, -2.1771e+01,  ..., -2.0907e+01,\n",
       "             1.9593e+00, -2.2233e+01],\n",
       "           [ 2.0688e+01, -3.3399e+01, -2.0001e+01,  ..., -8.6227e+00,\n",
       "            -2.6697e+01,  1.2888e+00],\n",
       "           [-1.9729e+01,  5.1690e-01,  9.8437e+00,  ...,  8.3385e+00,\n",
       "             1.9819e+01, -1.5859e+01],\n",
       "           ...,\n",
       "           [ 2.1852e+01, -1.1978e+01, -4.0928e+01,  ...,  1.0490e+00,\n",
       "            -2.2549e+01,  4.4806e+01],\n",
       "           [ 8.2454e+00,  9.6797e+00, -4.7854e+00,  ...,  1.4053e+01,\n",
       "            -1.4704e+01, -3.1102e+01],\n",
       "           [ 1.2226e+01,  2.3296e+01,  8.6078e-01,  ...,  3.2787e+01,\n",
       "             2.1104e+01, -3.4714e+00]],\n",
       "\n",
       "          [[-6.0582e+00,  4.7460e+01,  1.0296e+01,  ...,  2.9844e+01,\n",
       "            -9.5303e+00,  2.6007e+01],\n",
       "           [-9.1247e+00,  1.8553e+01,  1.2341e+00,  ..., -1.9020e+01,\n",
       "            -2.8622e+01, -3.1613e+01],\n",
       "           [ 2.9273e+01,  6.8169e+00,  1.4001e+01,  ..., -3.1977e+00,\n",
       "            -1.6819e+01, -4.3084e+00],\n",
       "           ...,\n",
       "           [-1.9398e+01,  1.9471e+01, -2.1960e+01,  ..., -1.0030e+01,\n",
       "            -4.5080e+01,  3.8009e+01],\n",
       "           [ 3.9487e+01, -1.2208e+01, -6.2321e+00,  ..., -8.0179e+00,\n",
       "            -1.2438e+01,  2.4806e+01],\n",
       "           [-1.7725e+01, -2.3344e+01, -9.9394e+00,  ...,  1.3042e+01,\n",
       "             2.2552e+01, -9.6741e+00]],\n",
       "\n",
       "          [[-1.7471e+01,  1.4914e+01, -1.4371e+01,  ..., -1.1043e+01,\n",
       "             3.5072e+01,  2.4643e+01],\n",
       "           [ 3.9464e+01, -3.0444e+01, -7.9941e+00,  ..., -1.3443e+01,\n",
       "             6.9515e+00,  5.5987e+00],\n",
       "           [-3.1306e+01, -8.6578e+00,  1.2478e+00,  ..., -2.3344e-01,\n",
       "            -1.2961e+00,  1.2098e+01],\n",
       "           ...,\n",
       "           [ 2.1710e+01, -3.5616e+01, -4.7083e+00,  ...,  1.1991e+00,\n",
       "            -3.0403e+01,  7.5907e+00],\n",
       "           [-1.5061e+01, -4.8471e+01,  2.8782e+01,  ...,  7.5877e+00,\n",
       "             1.7059e+01, -8.4924e+00],\n",
       "           [ 7.7232e+00, -2.7590e+01, -2.1477e+01,  ...,  2.1180e+01,\n",
       "             1.8627e+01, -9.7625e+00]]]],\n",
       "\n",
       "\n",
       "\n",
       "        ...,\n",
       "\n",
       "\n",
       "\n",
       "        [[[[-2.7130e+01, -2.2915e+01, -1.8498e+01,  ...,  1.8371e+00,\n",
       "             1.0636e+01,  4.1379e+00],\n",
       "           [-2.3977e+01, -1.3230e+01, -7.3330e+00,  ...,  3.4742e+00,\n",
       "            -2.6901e+00, -1.3161e+01],\n",
       "           [-1.1620e+01,  2.4289e+01,  1.0535e+01,  ...,  1.9432e+01,\n",
       "             4.5056e+00,  1.6505e+01],\n",
       "           ...,\n",
       "           [ 4.3697e+00, -2.9929e-01, -2.4537e+01,  ...,  4.7750e+00,\n",
       "             4.7462e+01, -1.5267e+01],\n",
       "           [-1.8421e+01,  4.1589e+00,  5.9645e+00,  ...,  1.6661e+01,\n",
       "            -1.5995e+01, -1.9210e+01],\n",
       "           [ 2.9617e+01,  2.5377e+01, -4.5354e+00,  ..., -2.7908e+01,\n",
       "            -3.5785e+01,  2.2551e+01]],\n",
       "\n",
       "          [[ 2.5576e+01,  5.3342e-01, -1.4441e+00,  ...,  1.1513e+01,\n",
       "            -4.1693e+01, -6.4107e+00],\n",
       "           [ 2.6286e+01,  9.7324e+00, -3.9693e+00,  ...,  3.5899e+01,\n",
       "            -2.2664e+00,  1.3940e+01],\n",
       "           [-2.2059e+01, -1.1281e+01, -8.8621e+00,  ..., -2.9630e+00,\n",
       "             7.7536e+00,  7.3092e+00],\n",
       "           ...,\n",
       "           [-3.5965e+00, -1.9637e+01,  8.0258e-01,  ...,  2.5395e+01,\n",
       "            -2.2247e+01,  4.7804e+01],\n",
       "           [ 1.3414e+01, -1.3282e+01,  2.4776e+01,  ...,  2.1792e+01,\n",
       "             1.3684e+01,  9.2705e+00],\n",
       "           [ 1.4753e+01, -5.0902e+00, -9.6245e+00,  ...,  2.8172e+01,\n",
       "             3.2863e+01,  3.4979e-01]],\n",
       "\n",
       "          [[ 2.0117e+01,  5.6794e-01, -3.6876e+01,  ...,  4.5119e+00,\n",
       "            -2.8270e-01,  3.7078e+01],\n",
       "           [ 1.7647e+01,  2.0353e+01, -1.8120e+01,  ...,  3.9788e+00,\n",
       "             2.2255e+00, -4.1274e+01],\n",
       "           [ 3.2148e+01,  1.4728e+01,  6.4598e+00,  ...,  7.9272e+00,\n",
       "            -8.8745e+00, -2.2774e+01],\n",
       "           ...,\n",
       "           [-3.6369e+00, -1.4202e+01, -1.6407e+01,  ..., -5.8392e+00,\n",
       "             1.4518e+01, -2.9914e+01],\n",
       "           [-1.8631e+01, -3.1858e+01, -8.0443e+00,  ...,  9.3801e-01,\n",
       "             1.0739e+01,  4.5816e+01],\n",
       "           [-8.0820e+00, -1.1628e+01, -6.6378e+00,  ..., -1.5257e+01,\n",
       "            -9.1153e+00,  3.9455e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 3.0189e+01,  1.8686e+01,  1.3485e+01,  ...,  2.6061e+01,\n",
       "            -1.3604e+01, -1.3571e+01],\n",
       "           [ 3.4426e+01,  1.9073e+01,  1.1854e+01,  ..., -1.1572e+01,\n",
       "            -1.2104e+01,  1.4135e+01],\n",
       "           [-2.0094e+01,  3.0158e+01, -1.3436e+01,  ...,  2.1547e+00,\n",
       "            -2.8294e+01, -4.7237e+00],\n",
       "           ...,\n",
       "           [ 4.4069e+00, -8.0673e+00,  3.5075e+00,  ..., -2.6624e+00,\n",
       "             6.4291e+00,  1.5950e+01],\n",
       "           [-9.3144e+00,  3.1295e+00,  1.3809e+01,  ..., -3.3851e+01,\n",
       "            -5.6221e+00,  7.4898e+00],\n",
       "           [-1.1300e+01, -4.5672e+00, -4.7933e+00,  ..., -5.6750e+00,\n",
       "            -1.5201e+01,  3.8842e+01]],\n",
       "\n",
       "          [[-2.5209e+01,  1.5433e+01, -4.9672e+00,  ..., -2.5028e+01,\n",
       "             3.1935e+00, -6.7257e+00],\n",
       "           [ 2.2399e-01,  5.8316e+00, -5.7261e+00,  ..., -7.3887e+00,\n",
       "             3.4698e+00, -6.4126e+00],\n",
       "           [-1.1838e+00,  2.8737e+01, -1.2900e+01,  ..., -2.6160e+01,\n",
       "            -5.2445e+00,  8.2405e+00],\n",
       "           ...,\n",
       "           [-7.4913e+00,  1.9577e+01,  1.3478e+01,  ..., -7.3794e+01,\n",
       "             2.7396e+01,  4.9113e+00],\n",
       "           [-1.9532e+01,  1.9049e+01, -2.8944e+00,  ..., -5.5132e+00,\n",
       "            -3.1940e+01,  3.7266e+01],\n",
       "           [-7.3198e+00,  1.1720e+01, -1.9900e+01,  ..., -2.2452e+01,\n",
       "            -1.6650e+01,  1.4704e+01]],\n",
       "\n",
       "          [[-3.7862e+01,  8.0010e+00,  1.3591e+01,  ...,  2.7818e+01,\n",
       "            -2.0590e+00, -1.2958e+01],\n",
       "           [ 1.8884e+01, -3.4968e+01, -3.3200e+01,  ...,  4.9572e+01,\n",
       "             1.7968e+01, -7.1428e+00],\n",
       "           [ 2.6226e+01,  1.9681e+01, -6.6407e-01,  ..., -1.8375e+00,\n",
       "             2.1821e+01,  2.2062e+01],\n",
       "           ...,\n",
       "           [-1.1630e+01, -2.0959e+01, -1.1527e+01,  ...,  3.0910e+01,\n",
       "             3.7698e+01,  2.5817e+00],\n",
       "           [ 1.0912e-01, -6.1707e+01, -2.2790e+01,  ..., -2.3916e+01,\n",
       "            -1.1426e+01, -3.8512e+01],\n",
       "           [ 7.1018e+00,  2.3024e+01,  1.2514e+01,  ...,  1.4540e+01,\n",
       "             2.2496e+01, -9.7467e+00]]],\n",
       "\n",
       "\n",
       "         [[[ 3.6885e+01, -3.3428e+00, -1.3569e+01,  ..., -2.9216e+00,\n",
       "            -1.1221e+01,  3.8055e+01],\n",
       "           [ 1.5173e+00, -3.4465e+01, -1.9823e+01,  ..., -3.3944e+01,\n",
       "             5.4335e+00,  2.5210e+01],\n",
       "           [-6.3798e+00, -4.3721e+01, -2.0345e+01,  ..., -2.0776e+01,\n",
       "            -3.9714e+00, -3.5080e+01],\n",
       "           ...,\n",
       "           [-1.2382e+01, -5.2781e+00,  5.1998e+00,  ..., -2.4669e+01,\n",
       "             5.0179e+01, -2.6696e+00],\n",
       "           [ 1.3930e+01,  2.4638e+01, -2.1044e+00,  ..., -2.3645e+01,\n",
       "             2.8380e+01,  3.2512e+01],\n",
       "           [-1.4165e+01, -1.1242e+01,  1.0402e+01,  ...,  2.7303e+01,\n",
       "             2.9004e+01, -1.1351e+01]],\n",
       "\n",
       "          [[ 4.6258e+01, -1.0221e+01,  8.6285e-01,  ..., -2.1610e+01,\n",
       "            -4.6229e+01, -1.4854e+00],\n",
       "           [ 1.1694e+01, -6.8530e+01, -1.5600e+01,  ..., -2.5947e+01,\n",
       "            -5.9978e+00, -1.3975e+00],\n",
       "           [ 8.9672e-01, -1.5384e+01, -1.8495e+00,  ...,  1.1677e+01,\n",
       "            -3.2328e+00, -2.3626e+01],\n",
       "           ...,\n",
       "           [-5.3587e+00, -8.8233e-01,  2.2881e+00,  ..., -3.3346e+01,\n",
       "            -2.3441e+00, -4.6294e+01],\n",
       "           [-4.5996e+01, -8.1098e+00, -3.0751e+00,  ..., -9.2706e+00,\n",
       "            -2.7867e+00,  1.2959e+01],\n",
       "           [-1.2977e+01, -3.8199e+01, -1.6434e+01,  ...,  3.6686e+01,\n",
       "             2.5145e+01,  2.1907e+01]],\n",
       "\n",
       "          [[ 8.5046e+00,  6.2034e+00, -2.2573e+01,  ..., -1.4455e+01,\n",
       "            -1.4182e-01,  2.5819e+01],\n",
       "           [-4.2051e+00,  1.7575e+01,  3.0268e-01,  ..., -9.2304e+00,\n",
       "            -2.1887e+01, -4.6179e-02],\n",
       "           [ 3.9113e+01, -1.9720e+00, -1.2902e+01,  ...,  2.7678e+00,\n",
       "            -1.6422e+00,  6.1980e+00],\n",
       "           ...,\n",
       "           [ 6.9291e+01,  1.2638e+01, -2.8012e+00,  ..., -8.9321e-01,\n",
       "            -3.2651e+01,  1.6630e+01],\n",
       "           [ 1.4434e+01, -1.4151e+01,  7.8130e+00,  ..., -7.9955e+00,\n",
       "             3.8320e+01,  2.1594e+01],\n",
       "           [ 1.1402e+01,  4.5469e+01,  7.0646e+00,  ...,  1.3504e+00,\n",
       "             1.6054e+01, -1.1251e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-3.2112e+01, -5.7590e-01,  2.3649e+01,  ..., -1.5298e+01,\n",
       "             3.3538e+01,  7.9155e+00],\n",
       "           [-1.6257e+01, -2.7329e+01, -5.1509e+00,  ...,  5.3087e+01,\n",
       "             5.5548e-01,  3.5855e+01],\n",
       "           [-1.5838e+01,  5.0716e+00,  1.1362e+01,  ..., -3.1705e+01,\n",
       "            -9.6118e-01, -3.2126e+01],\n",
       "           ...,\n",
       "           [-1.2396e+01, -1.2792e+01,  1.2497e+01,  ...,  1.4529e+01,\n",
       "             5.5659e+01,  9.8563e+00],\n",
       "           [ 5.2705e+00,  3.6753e+01,  1.5314e+01,  ..., -1.7652e+01,\n",
       "             4.0110e+01, -5.9128e+00],\n",
       "           [-7.1394e+00,  1.3501e+01, -4.2579e+01,  ...,  1.0438e+00,\n",
       "             1.7764e+01,  2.0437e+01]],\n",
       "\n",
       "          [[-2.0622e+01,  9.8667e+00,  9.2626e+00,  ..., -6.0322e+00,\n",
       "             1.2164e+01, -3.5467e+00],\n",
       "           [ 9.5854e+00,  1.1187e+00,  2.1465e+01,  ...,  1.5884e+01,\n",
       "            -3.3075e+01, -9.3666e+00],\n",
       "           [ 4.6416e+00, -1.3990e+01,  1.3482e+01,  ...,  1.5703e+01,\n",
       "             3.6060e+01,  8.5004e+00],\n",
       "           ...,\n",
       "           [-2.3056e+01,  3.1411e+01,  8.4657e+00,  ..., -1.4623e+01,\n",
       "             5.9808e+01,  2.8285e+01],\n",
       "           [ 1.1471e+01, -1.4447e+01, -2.4314e+01,  ..., -7.6461e+00,\n",
       "             1.1625e+01, -1.0626e+01],\n",
       "           [ 1.2880e+01, -2.3783e+01, -8.2909e+00,  ...,  2.6261e+01,\n",
       "             3.2021e+01, -1.2927e+01]],\n",
       "\n",
       "          [[ 8.4598e-01, -1.4663e+01,  3.1721e+01,  ..., -4.2492e+01,\n",
       "            -6.3248e+00,  3.1538e+01],\n",
       "           [ 4.3283e+01, -4.4515e+00, -3.2675e+00,  ...,  3.6360e+01,\n",
       "            -2.0102e+01, -1.4052e+01],\n",
       "           [ 1.2203e+01, -1.9334e+00, -1.6144e+01,  ...,  1.4077e+01,\n",
       "            -1.9595e+01, -1.5254e+01],\n",
       "           ...,\n",
       "           [-1.6307e+01,  1.6687e+01,  2.1700e+01,  ...,  1.3046e+01,\n",
       "            -1.0132e+01,  2.6080e+01],\n",
       "           [-1.4888e+01,  1.8306e+01, -8.3247e+00,  ..., -2.2317e+01,\n",
       "             2.7110e+00,  2.8769e+01],\n",
       "           [ 1.7789e+01, -2.3700e+01, -6.2985e+00,  ..., -4.5204e+00,\n",
       "            -2.4928e+01, -2.1249e+01]]],\n",
       "\n",
       "\n",
       "         [[[-1.1350e+01, -7.5865e+00, -1.3930e+01,  ..., -5.0725e+01,\n",
       "            -7.4271e-01, -2.5646e+00],\n",
       "           [-3.3496e+01,  2.8124e+01,  5.8891e+00,  ...,  5.3632e+00,\n",
       "            -1.8967e+00, -2.3720e+01],\n",
       "           [ 1.9987e+01,  3.9284e+01,  6.4941e+00,  ..., -1.2095e+01,\n",
       "             4.6321e+01,  1.2924e+01],\n",
       "           ...,\n",
       "           [ 1.0897e+01,  1.0528e+01,  3.1195e+00,  ..., -3.8409e+00,\n",
       "             2.8893e+01, -1.2911e+01],\n",
       "           [ 2.3934e+01, -3.9692e+01,  4.9047e+01,  ...,  4.0539e+01,\n",
       "            -7.2398e+00, -1.3389e+01],\n",
       "           [ 7.4860e+00, -2.0137e+01,  1.2901e+01,  ..., -4.1616e+01,\n",
       "             2.4120e+01, -7.3679e-01]],\n",
       "\n",
       "          [[-2.9696e+01, -1.5183e+01, -2.7970e+01,  ..., -2.6919e+01,\n",
       "            -1.2601e+01,  1.1340e+01],\n",
       "           [ 1.7188e+00,  1.1399e+01,  3.6826e+01,  ...,  2.5719e+01,\n",
       "            -7.6723e+00, -1.5593e+00],\n",
       "           [-3.5032e+01,  1.7591e+01, -1.2191e+01,  ...,  2.8088e+00,\n",
       "             1.6910e+01,  1.4464e+01],\n",
       "           ...,\n",
       "           [-1.0372e+01, -1.7438e+00,  1.0164e+01,  ...,  2.3447e+01,\n",
       "             2.3354e+00, -4.7488e+00],\n",
       "           [ 3.5461e+00,  2.3565e+01,  6.2010e+00,  ...,  3.1250e+01,\n",
       "             3.2882e+01, -5.4271e+00],\n",
       "           [ 5.5337e+00, -1.7013e+01,  9.0072e+00,  ...,  4.2528e-02,\n",
       "             9.4208e+00,  3.8036e-01]],\n",
       "\n",
       "          [[-2.2983e+01, -3.5339e+01,  2.9397e+00,  ..., -1.1685e+01,\n",
       "            -2.1706e+01, -7.7240e+00],\n",
       "           [-1.2300e+01,  8.2909e+00, -1.5358e+01,  ...,  5.4026e+00,\n",
       "             1.8606e+01, -2.0302e+01],\n",
       "           [-5.2914e+01, -3.0156e+01,  3.0431e+00,  ..., -2.9996e+01,\n",
       "             2.3266e+00,  3.0597e+01],\n",
       "           ...,\n",
       "           [ 2.4976e+00, -3.2478e+00,  1.7256e+01,  ..., -1.8225e+01,\n",
       "            -7.8984e-02, -4.0398e+01],\n",
       "           [-8.2319e+00, -9.3368e+00, -1.2890e+01,  ...,  2.4305e+00,\n",
       "             9.4313e+00,  2.2328e+01],\n",
       "           [-1.0471e+01,  5.3421e+00,  4.9455e+01,  ...,  2.1556e+00,\n",
       "             2.3312e+01,  1.1524e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 2.2086e+01,  1.7474e+01,  2.6272e+01,  ..., -1.4070e+01,\n",
       "            -1.2626e+00,  2.3842e+01],\n",
       "           [-1.0553e+01, -2.1158e+01, -5.2176e+00,  ...,  2.1155e+01,\n",
       "            -9.2393e+00,  3.0719e+01],\n",
       "           [-1.1587e+01, -3.3452e+01,  4.5400e-01,  ..., -2.5314e+01,\n",
       "            -1.9107e+01, -1.4265e+01],\n",
       "           ...,\n",
       "           [-3.9780e+00,  6.4645e+00, -2.1779e+01,  ...,  2.6843e+01,\n",
       "             2.1143e+01, -1.3645e+01],\n",
       "           [ 3.4056e+01, -5.1760e-01,  1.7449e+01,  ...,  1.8149e+01,\n",
       "            -8.9773e+00,  2.2686e+01],\n",
       "           [ 9.1506e+00,  2.7030e+01,  1.9028e+01,  ..., -9.7755e+00,\n",
       "            -1.4638e+01,  2.3711e+01]],\n",
       "\n",
       "          [[-7.0193e+00,  2.6813e+01,  1.0723e+01,  ...,  2.9656e+01,\n",
       "             1.3922e+01, -2.4466e+01],\n",
       "           [ 1.6023e+01,  4.6124e+01, -3.2540e+01,  ..., -4.5671e+00,\n",
       "            -1.2952e+00, -1.4761e+01],\n",
       "           [-8.3178e+00, -2.0067e+01, -5.0924e+00,  ..., -2.8473e+01,\n",
       "             3.6876e+01,  3.0744e+01],\n",
       "           ...,\n",
       "           [-1.2922e+01,  1.7018e+01, -2.0360e+01,  ..., -3.9542e+00,\n",
       "             4.8514e+00,  1.6874e+00],\n",
       "           [ 3.2087e+01,  3.4601e+01,  1.6447e+01,  ..., -1.1684e+01,\n",
       "             1.3706e+01, -2.2386e+01],\n",
       "           [ 6.5091e+00, -3.4973e+01, -6.3920e+00,  ...,  4.6205e+00,\n",
       "             4.4827e+01,  1.1829e+01]],\n",
       "\n",
       "          [[-3.2031e+01,  1.0667e+01,  8.3395e+00,  ..., -4.9289e+00,\n",
       "            -9.4416e+00, -4.2556e+01],\n",
       "           [ 2.7135e+01, -3.9032e+01, -2.2187e+00,  ...,  1.6952e+01,\n",
       "             7.5840e+00, -2.1413e+01],\n",
       "           [-3.9371e+00, -6.2280e+00,  9.2477e+00,  ..., -2.5487e+01,\n",
       "             2.9629e+01, -2.7614e+00],\n",
       "           ...,\n",
       "           [-1.8885e+01, -1.8007e+01,  3.4703e+00,  ...,  5.5814e-01,\n",
       "             3.8331e+01,  1.2374e+01],\n",
       "           [-8.2191e+00, -1.9457e+00, -5.8367e-01,  ...,  5.2656e+01,\n",
       "            -1.6631e+01,  5.9999e+00],\n",
       "           [ 6.5494e+00,  9.1639e+00,  3.6604e+01,  ..., -4.6917e+01,\n",
       "            -2.8506e+01,  4.2144e+00]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[ 3.5108e+00,  3.8882e+01, -3.6600e+00,  ..., -3.2138e+01,\n",
       "             3.4057e+01, -1.5524e+01],\n",
       "           [ 1.2887e+01, -1.6376e+01,  5.2870e+00,  ..., -2.5566e+00,\n",
       "             5.6420e+00, -3.4084e+01],\n",
       "           [ 9.9845e+00,  2.2827e+01,  2.3015e+01,  ..., -1.3598e+01,\n",
       "             2.5819e+01,  1.9674e+00],\n",
       "           ...,\n",
       "           [-3.1629e+01, -1.8733e+01, -1.5380e+00,  ...,  5.3376e+00,\n",
       "             2.3183e+01,  1.9020e+01],\n",
       "           [-4.8309e+01,  1.9398e+01, -1.6642e+01,  ..., -2.2076e+01,\n",
       "            -3.7022e+00,  1.4360e+01],\n",
       "           [-1.8696e+01,  1.9790e+00,  1.0245e+01,  ...,  8.6026e+00,\n",
       "             2.9360e+01, -1.2254e+01]],\n",
       "\n",
       "          [[ 1.2285e+01,  3.4931e+00, -1.0709e+01,  ..., -1.3611e+01,\n",
       "             7.7935e+00,  2.2060e+01],\n",
       "           [ 3.8866e+00,  2.3912e+01,  4.5071e+01,  ..., -8.4038e+00,\n",
       "            -1.6069e+01,  1.2303e+00],\n",
       "           [-1.4684e+01, -1.4205e+01, -3.3357e+01,  ..., -2.7346e+01,\n",
       "             5.8387e+01,  2.5845e+00],\n",
       "           ...,\n",
       "           [ 6.2127e+00,  1.4990e+01, -3.7737e+01,  ...,  4.7032e+01,\n",
       "             2.9214e+01, -1.9375e+01],\n",
       "           [ 1.6295e+01, -1.2205e+01, -1.2073e+01,  ...,  2.0097e+01,\n",
       "            -5.6530e+01, -1.6397e+01],\n",
       "           [ 2.7994e+01, -1.8464e+01,  3.5139e+01,  ...,  3.3527e+01,\n",
       "             1.3383e+00,  5.8364e+00]],\n",
       "\n",
       "          [[-7.0807e+00,  1.0789e+01,  2.9082e+01,  ...,  6.9624e+00,\n",
       "            -7.3562e+00,  1.4204e+01],\n",
       "           [ 2.1048e+00, -1.0951e+01,  8.1420e+00,  ...,  2.0856e+01,\n",
       "            -1.0750e+01, -1.6519e+01],\n",
       "           [-3.6913e+01, -2.3703e+00, -6.8464e+00,  ..., -9.2205e+00,\n",
       "             4.8989e+01, -3.0095e+01],\n",
       "           ...,\n",
       "           [ 2.9426e+01,  3.4762e+00,  3.2134e+01,  ...,  3.4596e+01,\n",
       "            -1.4165e+01,  7.6785e+00],\n",
       "           [-1.3250e+01,  3.0846e+01,  3.1212e+01,  ...,  9.4096e+00,\n",
       "             1.3820e+01, -1.6579e+01],\n",
       "           [ 4.0542e+00,  3.6783e+01, -1.1574e+01,  ..., -9.8842e+00,\n",
       "             1.3825e+01, -1.6440e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.6368e+00,  7.3478e+00, -2.8932e+00,  ...,  1.1114e+01,\n",
       "             1.3781e+01, -1.9631e+01],\n",
       "           [-2.5634e+01,  2.1757e+00, -1.1994e+01,  ..., -2.1542e+01,\n",
       "            -4.0315e+00,  2.7688e+01],\n",
       "           [ 1.1813e+00,  2.7172e+01, -1.8003e+01,  ...,  8.8186e+00,\n",
       "             1.4858e+01, -1.2539e+01],\n",
       "           ...,\n",
       "           [-6.3264e+00,  3.3599e+01,  2.6706e+00,  ..., -2.1414e+01,\n",
       "            -3.3969e+01,  3.5127e-01],\n",
       "           [-5.2705e+00,  1.3068e+01,  7.0752e+00,  ...,  4.2331e+01,\n",
       "             2.3055e+01,  2.2911e+01],\n",
       "           [-1.6362e+01,  4.9797e+00,  1.1459e+00,  ...,  2.6334e+00,\n",
       "            -1.9654e+01, -8.0747e+00]],\n",
       "\n",
       "          [[-1.1633e+01,  1.5251e+01,  2.8302e+01,  ...,  2.8089e+01,\n",
       "            -1.8020e+01, -3.4904e+01],\n",
       "           [ 3.1209e+01,  3.6501e+01, -2.5103e+01,  ...,  3.3057e+01,\n",
       "            -1.7064e+00,  6.5248e+00],\n",
       "           [ 3.1036e+01,  1.0549e+01,  2.7664e+00,  ..., -1.2760e+00,\n",
       "             3.7782e+01,  2.7019e+01],\n",
       "           ...,\n",
       "           [-9.4766e+00,  1.4554e+00,  3.4973e-01,  ...,  1.0745e+01,\n",
       "            -4.9197e+00,  3.0914e+01],\n",
       "           [-1.4660e+01, -3.3673e-02,  3.1713e+00,  ..., -2.6755e+00,\n",
       "             3.5751e+01, -1.5893e+01],\n",
       "           [ 2.3416e+00,  9.6737e+00,  1.3979e+01,  ..., -2.0007e+01,\n",
       "            -6.1903e+00,  4.2361e+00]],\n",
       "\n",
       "          [[ 1.8830e+01, -1.8389e+01, -1.5269e-01,  ..., -3.3635e+01,\n",
       "            -7.9576e+00, -7.0320e+00],\n",
       "           [ 1.3992e+01,  5.2786e+00,  1.5758e+01,  ..., -9.7826e+00,\n",
       "            -1.5944e+01,  2.1203e+01],\n",
       "           [ 2.2544e+01, -1.2853e+01, -1.7108e+01,  ...,  6.4083e+00,\n",
       "            -1.1918e+01,  1.9400e+01],\n",
       "           ...,\n",
       "           [-3.2291e+01, -1.0123e+01,  4.8062e+01,  ..., -2.1907e+01,\n",
       "            -1.3533e+01,  1.4746e+01],\n",
       "           [ 1.0336e+01,  8.4318e+00,  2.7416e+01,  ...,  5.3236e+00,\n",
       "             2.1717e+01,  2.3599e+01],\n",
       "           [-1.5202e+01, -1.0939e+01, -2.5039e+00,  ..., -1.6807e+01,\n",
       "            -7.4219e+00,  2.9736e+01]]],\n",
       "\n",
       "\n",
       "         [[[ 1.2325e+01,  4.0184e+01, -9.3950e+00,  ..., -2.3831e+01,\n",
       "            -7.9156e+00,  2.2423e+01],\n",
       "           [-1.2934e+00,  1.6764e+01, -1.1456e+01,  ..., -2.6206e+01,\n",
       "             6.3215e+00,  5.2220e+00],\n",
       "           [ 4.4554e+00,  4.5865e+01,  1.1386e+01,  ..., -1.6529e+01,\n",
       "            -5.7649e+00, -1.4676e+01],\n",
       "           ...,\n",
       "           [ 2.7819e+01,  1.7520e+01, -1.7287e+01,  ...,  1.9864e+01,\n",
       "             3.1303e+01,  1.1151e+00],\n",
       "           [-2.2484e+01,  1.0905e+01, -1.3239e+01,  ...,  1.3284e+01,\n",
       "             7.7347e+00, -1.3075e+01],\n",
       "           [ 1.1702e+01, -3.2814e+01,  4.1573e+01,  ...,  6.1924e+00,\n",
       "            -7.0705e+00, -4.5960e+00]],\n",
       "\n",
       "          [[-8.5205e+00,  4.7540e+01, -3.7965e+01,  ..., -9.7426e+00,\n",
       "            -1.5645e+01,  2.6282e+00],\n",
       "           [-9.1701e+00,  7.8428e-01, -1.3975e+01,  ..., -1.7761e+01,\n",
       "             2.1514e+00, -3.5897e+01],\n",
       "           [-1.3745e+01, -7.4665e+00, -2.9346e+01,  ..., -4.6168e+00,\n",
       "            -4.5603e+00, -9.4226e+00],\n",
       "           ...,\n",
       "           [ 2.4941e+01, -8.8815e+00, -2.1016e+01,  ...,  2.6100e+01,\n",
       "             1.7358e+01,  2.6324e+00],\n",
       "           [-4.5007e+00,  1.3212e+00, -7.4248e+00,  ...,  1.3280e+01,\n",
       "            -1.7827e+00,  2.6364e+01],\n",
       "           [-1.9339e+01,  9.9451e+00, -5.3835e+00,  ..., -8.1993e+00,\n",
       "            -1.1266e+01, -1.9321e+01]],\n",
       "\n",
       "          [[-1.0772e+01, -1.3686e+01,  4.0368e+00,  ..., -1.6072e+01,\n",
       "            -1.5967e+01,  7.4987e+00],\n",
       "           [ 1.5420e+01, -1.0872e+01, -2.2951e+01,  ..., -2.1306e+00,\n",
       "            -4.4045e+00, -1.2958e+01],\n",
       "           [-5.0943e+00,  4.2127e+01, -4.0826e+01,  ...,  4.5569e+00,\n",
       "            -1.9163e+01, -7.0626e+00],\n",
       "           ...,\n",
       "           [-3.4011e+01, -4.6352e+01,  2.1193e+00,  ...,  9.3740e+00,\n",
       "             2.9470e+01, -2.0885e+01],\n",
       "           [-2.5889e+01,  1.6477e+01,  3.2385e+01,  ..., -3.5359e+01,\n",
       "            -2.5140e+00,  3.4492e+01],\n",
       "           [ 4.5371e+00, -7.5780e+00, -1.9572e+00,  ...,  2.2427e+00,\n",
       "            -2.2794e+01,  2.2019e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.1477e+01,  1.2894e+01,  2.2272e+01,  ..., -4.1921e+00,\n",
       "            -1.5823e-01, -7.9963e+00],\n",
       "           [-3.3616e+01,  1.7095e+01,  2.0006e+01,  ...,  2.3168e+01,\n",
       "             2.1925e+01, -1.5437e+01],\n",
       "           [ 1.3948e+01,  1.6911e+01,  6.3424e+00,  ...,  1.1081e+01,\n",
       "            -3.3943e+01,  1.2748e+01],\n",
       "           ...,\n",
       "           [-1.0860e+00,  3.5518e+01, -1.2195e+01,  ...,  1.3275e+01,\n",
       "            -2.8571e+01, -1.9862e+01],\n",
       "           [ 2.3917e+01,  4.9550e+00,  2.0527e+01,  ..., -2.2527e+01,\n",
       "            -1.6816e+01,  2.2333e+01],\n",
       "           [-2.4893e+01,  4.0204e+01,  3.5227e+01,  ..., -1.9637e+01,\n",
       "             8.1903e+00,  4.2729e+00]],\n",
       "\n",
       "          [[ 1.0744e+01, -9.9863e+00,  1.3161e+01,  ...,  1.5427e+00,\n",
       "            -6.3433e+01,  2.7967e+01],\n",
       "           [ 2.2237e+01, -4.5109e+00,  3.8928e-01,  ..., -2.9616e+01,\n",
       "             2.9735e+01, -2.0192e+01],\n",
       "           [ 1.8887e+01,  1.4102e+01,  4.0163e+01,  ..., -3.3927e+01,\n",
       "             2.8817e+01,  5.7192e+00],\n",
       "           ...,\n",
       "           [ 1.0751e+01,  8.1149e+00,  3.5104e+01,  ...,  5.2251e-01,\n",
       "            -1.6271e+01, -1.6856e+01],\n",
       "           [-1.0537e+01,  1.8905e+01,  4.3332e+01,  ..., -2.2881e+01,\n",
       "             3.5320e+01, -3.8640e+01],\n",
       "           [ 1.9347e+01,  6.9770e-01,  1.9393e+00,  ...,  2.0412e+01,\n",
       "             1.7747e+01, -4.1966e+01]],\n",
       "\n",
       "          [[-1.5000e+01, -1.5695e+00,  1.2436e+01,  ...,  3.2321e+01,\n",
       "            -2.4422e+01,  1.7097e+01],\n",
       "           [-2.6853e+01,  1.5611e+01, -2.7462e+01,  ..., -2.6314e+01,\n",
       "             1.1468e+01,  6.2157e+00],\n",
       "           [-1.4115e+01,  1.4319e+01, -1.4083e+01,  ..., -6.0930e+00,\n",
       "             3.7603e+01, -2.1537e+01],\n",
       "           ...,\n",
       "           [-9.8740e+00,  5.7247e+00, -2.7802e+01,  ..., -1.5912e+00,\n",
       "             1.0047e+01,  1.3663e+01],\n",
       "           [ 4.0342e+00,  2.2552e+00, -1.0172e+01,  ...,  5.9096e+00,\n",
       "             2.3811e+00, -5.5223e+00],\n",
       "           [ 3.4248e+00, -1.8438e+01, -1.9338e+01,  ..., -3.0329e+01,\n",
       "            -1.2374e+01,  3.1010e+01]]],\n",
       "\n",
       "\n",
       "         [[[ 1.5748e+01, -2.0534e+01, -1.6158e+01,  ...,  1.5981e+01,\n",
       "            -2.3986e+01, -1.3280e+01],\n",
       "           [-2.3881e+01, -7.1962e+00, -4.5910e+01,  ...,  1.7826e+00,\n",
       "             1.9285e+00, -1.0298e+01],\n",
       "           [-4.0843e+01,  2.9874e+01,  2.7561e+01,  ...,  1.0833e+01,\n",
       "            -2.5811e+01, -2.1089e+01],\n",
       "           ...,\n",
       "           [ 1.5639e+01, -9.1169e+00,  2.2565e+01,  ..., -3.8667e+00,\n",
       "            -1.0272e+00, -8.0117e+00],\n",
       "           [ 1.1432e+01,  1.5410e+01,  3.1909e+01,  ...,  1.1304e+01,\n",
       "            -9.2372e+00,  2.0295e+01],\n",
       "           [-1.1460e+01,  1.1567e+01,  3.1682e+01,  ..., -2.0967e+01,\n",
       "            -2.1099e+01,  2.9166e+01]],\n",
       "\n",
       "          [[-1.4345e+01, -1.3061e+01, -7.1763e+00,  ..., -1.8072e+00,\n",
       "             3.7038e+01, -7.5455e+00],\n",
       "           [-3.5609e+01,  2.3221e+01,  1.2817e+01,  ...,  2.9613e+01,\n",
       "            -1.6965e+01,  1.2105e+01],\n",
       "           [-2.9796e+01, -6.2371e+00, -1.3742e+01,  ..., -7.5554e+00,\n",
       "             1.9447e+01,  1.5172e+01],\n",
       "           ...,\n",
       "           [-3.0927e+01,  2.1293e+01, -1.3091e+01,  ...,  2.3229e+00,\n",
       "            -2.9342e+01,  2.4936e+01],\n",
       "           [ 3.3919e-01,  2.2423e+01, -3.7121e+01,  ...,  1.6166e+01,\n",
       "             3.0187e+01, -2.5354e-01],\n",
       "           [ 3.2289e+01,  1.2921e+01,  1.1605e+01,  ..., -1.9155e+01,\n",
       "             2.5928e+01, -3.3896e+01]],\n",
       "\n",
       "          [[-1.4460e+01,  1.4795e+01,  3.7497e+01,  ..., -1.2697e+01,\n",
       "            -1.9040e+01, -2.7114e+01],\n",
       "           [ 2.0193e+01, -2.4967e+00,  1.0813e+01,  ..., -1.5956e+01,\n",
       "             1.5550e+01,  1.3469e+00],\n",
       "           [-3.9485e+01,  4.9889e+00, -3.0499e+01,  ...,  1.6055e+01,\n",
       "            -6.1124e+00,  9.3078e+00],\n",
       "           ...,\n",
       "           [-2.0096e+01,  1.5330e+01, -6.6594e+00,  ..., -5.7918e-01,\n",
       "             8.7362e+00, -5.9170e+01],\n",
       "           [ 1.7314e+01,  2.0870e+01, -5.6782e+00,  ..., -2.5137e+01,\n",
       "             1.1614e+01,  3.8663e+00],\n",
       "           [ 9.8297e+00, -7.9001e+00,  2.8139e+01,  ...,  1.5656e+00,\n",
       "             2.7053e+01,  2.4401e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 2.7876e+01, -2.8181e+01, -1.1635e+01,  ...,  5.6076e+00,\n",
       "            -5.5858e+00, -8.8696e+00],\n",
       "           [ 3.5664e+01, -2.3476e+01, -3.0561e+01,  ..., -3.2472e+01,\n",
       "             1.3389e+01, -1.7355e+01],\n",
       "           [-5.4240e+00,  6.9940e+00, -1.9854e+01,  ...,  6.1742e+00,\n",
       "            -3.0068e+01,  1.5382e+00],\n",
       "           ...,\n",
       "           [ 1.2072e+01,  1.6111e+01,  1.8457e+01,  ...,  4.9701e+01,\n",
       "            -3.4229e+01,  1.0011e+01],\n",
       "           [-4.1092e+00, -2.0969e+01,  3.4493e+01,  ...,  1.6345e+01,\n",
       "            -4.7087e+00, -2.0934e+01],\n",
       "           [-2.3139e+01,  9.3492e+00,  1.7565e+01,  ..., -1.3405e+01,\n",
       "             3.1309e+01,  3.3745e+01]],\n",
       "\n",
       "          [[-3.4921e+01,  1.9806e+01, -1.4421e+01,  ...,  1.1296e+01,\n",
       "             2.6724e+01,  9.9951e+00],\n",
       "           [ 2.1182e+00,  3.9450e+01,  3.7021e+01,  ...,  3.9728e+01,\n",
       "            -3.0443e-01, -9.4959e+00],\n",
       "           [ 1.5090e+01,  1.7423e+01,  1.4015e+01,  ..., -4.3449e+01,\n",
       "             1.1486e+01, -2.5680e+01],\n",
       "           ...,\n",
       "           [ 2.0143e+01, -5.7107e+00,  8.3884e+00,  ...,  2.3360e+01,\n",
       "            -7.9288e+00,  2.4065e+01],\n",
       "           [-3.2227e+01,  3.0143e+01, -8.5125e+00,  ..., -6.3148e+00,\n",
       "            -8.4090e+00,  7.5500e+00],\n",
       "           [ 7.8999e+00, -1.2499e+01,  2.1393e+01,  ...,  4.6779e+01,\n",
       "            -3.8922e+01,  3.1693e+01]],\n",
       "\n",
       "          [[ 1.3637e+01, -3.0327e+01, -3.6881e+01,  ..., -4.1346e+00,\n",
       "            -7.6838e+00, -4.6041e+01],\n",
       "           [-1.3668e+01, -1.4660e+01, -1.5440e+01,  ..., -3.4477e+01,\n",
       "            -1.5385e+00,  1.6085e+01],\n",
       "           [-7.8328e+00,  2.3442e+01,  3.3112e+01,  ..., -2.6376e-01,\n",
       "            -1.4078e+00,  1.1754e+01],\n",
       "           ...,\n",
       "           [-2.5638e+00,  5.6834e+00, -1.9448e+01,  ..., -1.0017e+01,\n",
       "             2.4589e+01,  3.7835e+01],\n",
       "           [ 5.1555e+00, -1.2926e-01, -1.8139e+01,  ...,  5.0324e+00,\n",
       "             1.1239e+01, -3.6904e+01],\n",
       "           [ 5.1100e+00, -2.1240e+01,  5.1399e+01,  ...,  1.9808e+01,\n",
       "            -1.4875e+01,  1.9099e+00]]]],\n",
       "\n",
       "\n",
       "\n",
       "        [[[[-2.2819e+01, -1.6098e+01, -8.3821e+00,  ...,  9.5860e+00,\n",
       "             5.3316e+01,  6.0139e+00],\n",
       "           [ 5.6897e+00,  4.4963e-01, -1.2804e+01,  ...,  2.6708e+01,\n",
       "             7.9498e+00, -1.3203e+00],\n",
       "           [ 1.3614e+01, -1.5282e+01, -5.5709e+00,  ...,  1.6228e+01,\n",
       "             1.8543e+01, -3.2149e+01],\n",
       "           ...,\n",
       "           [ 3.9562e+01,  2.0570e+01,  2.3866e+01,  ..., -3.0440e+01,\n",
       "             1.3993e+01,  4.0334e+01],\n",
       "           [-6.9278e+00,  2.5794e+01, -1.0032e+01,  ..., -4.0387e+00,\n",
       "            -2.3292e+01, -1.2255e+00],\n",
       "           [ 2.1968e+01, -2.1021e+01, -2.0307e+00,  ...,  2.6209e+01,\n",
       "             3.0571e+01, -6.9895e+00]],\n",
       "\n",
       "          [[-9.0065e+00,  1.4333e+01,  2.9229e+01,  ..., -1.6236e+01,\n",
       "            -1.5419e+01, -1.0561e+01],\n",
       "           [ 2.6033e+01, -3.8797e+01,  3.3688e+01,  ...,  1.0161e+01,\n",
       "            -5.7208e+00, -1.8254e+01],\n",
       "           [ 2.1037e+01,  9.0413e+00, -6.2619e+00,  ..., -3.1034e+00,\n",
       "            -2.0099e+01, -2.3689e+01],\n",
       "           ...,\n",
       "           [-8.3596e+00, -3.4519e+01,  1.0594e+01,  ...,  6.8690e+00,\n",
       "            -8.5187e+00,  1.6968e+01],\n",
       "           [ 3.0088e+01,  2.0574e+01,  1.3263e+01,  ...,  8.9952e+00,\n",
       "             1.9769e+01, -1.8208e+01],\n",
       "           [-3.0165e+00,  8.7197e+00,  2.0016e+00,  ..., -4.2625e+01,\n",
       "             2.0052e+01, -1.2828e+01]],\n",
       "\n",
       "          [[ 3.1256e-01,  4.2763e+01,  2.4397e+01,  ...,  2.3061e+01,\n",
       "            -2.6887e-01, -1.3027e+01],\n",
       "           [ 4.1214e+01,  6.0394e+01,  7.6991e+00,  ...,  4.3213e-01,\n",
       "            -2.0711e+01,  3.8852e+00],\n",
       "           [-2.2854e+01,  2.9751e+01, -3.7610e+01,  ...,  1.3218e+01,\n",
       "            -2.0025e+01,  2.0013e+01],\n",
       "           ...,\n",
       "           [ 1.4393e+00, -3.5551e+01,  2.1201e+01,  ...,  4.3483e+01,\n",
       "            -1.7619e+01,  5.1605e-01],\n",
       "           [-5.0980e+00,  1.0935e+01,  2.0417e+01,  ..., -1.4574e+01,\n",
       "             3.3353e+01,  2.1777e+00],\n",
       "           [-8.3085e+00,  2.2065e+01, -1.1546e+01,  ..., -9.6978e-01,\n",
       "            -1.8465e+01,  7.7378e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 4.2665e+00,  9.2896e-01, -5.0853e+00,  ..., -1.6726e+01,\n",
       "             9.9855e+00,  4.2327e+01],\n",
       "           [ 2.0777e+01, -2.6428e+01, -1.7868e+01,  ..., -3.5837e+01,\n",
       "             2.5430e+01, -1.2321e+01],\n",
       "           [-1.1663e+01,  3.6314e+01,  1.7678e+01,  ...,  8.6287e+00,\n",
       "             1.8225e+01,  2.8914e+00],\n",
       "           ...,\n",
       "           [-2.2984e+01,  1.9611e+01, -2.2659e+01,  ..., -2.9931e+01,\n",
       "            -2.0166e+01, -7.9965e+00],\n",
       "           [ 1.7493e+01,  3.2188e+01, -3.2586e+01,  ...,  8.4246e+00,\n",
       "             2.1148e+01, -6.4434e+00],\n",
       "           [ 2.1973e+01,  8.7700e+00, -6.1999e+00,  ...,  2.3791e+01,\n",
       "             3.9833e+01, -2.7618e+01]],\n",
       "\n",
       "          [[ 1.3181e+01,  1.7534e+01, -1.8601e+01,  ..., -5.4275e+01,\n",
       "            -4.3098e+00, -1.6335e+01],\n",
       "           [-4.6801e+00, -2.0434e+00, -1.4208e+01,  ...,  4.0821e+00,\n",
       "             2.8309e+01,  1.7860e+01],\n",
       "           [ 1.3205e+01, -2.2521e+01, -2.5372e+01,  ..., -3.3081e+01,\n",
       "            -1.8854e+01, -1.2499e+01],\n",
       "           ...,\n",
       "           [ 1.6689e+01, -4.4791e+01, -7.3506e+00,  ..., -1.8428e+00,\n",
       "            -3.7223e+01, -3.9347e+00],\n",
       "           [-3.8843e+01, -1.9464e+01, -2.9111e+01,  ...,  3.0454e+00,\n",
       "             1.1935e+01,  1.3021e+00],\n",
       "           [-1.1563e+01, -8.9616e+00, -4.1343e+01,  ..., -2.2227e+01,\n",
       "             7.6666e+00, -1.4600e+01]],\n",
       "\n",
       "          [[-2.5868e+01, -3.8765e+01, -1.8512e+01,  ..., -9.4769e+00,\n",
       "            -1.3500e+01, -2.6915e+00],\n",
       "           [ 2.5780e+01, -2.4736e+00,  3.6871e+01,  ...,  3.4709e+01,\n",
       "             1.7573e+01,  5.8095e+00],\n",
       "           [ 3.7170e+01, -2.5051e+01, -7.4734e+00,  ...,  1.7020e+01,\n",
       "             1.9965e+01, -1.8919e+01],\n",
       "           ...,\n",
       "           [ 2.0599e+00, -1.3018e+01, -6.8823e+00,  ...,  2.1927e+01,\n",
       "            -1.1236e+01, -1.2911e+01],\n",
       "           [ 1.6236e+01,  2.2693e+01, -9.2226e+00,  ...,  3.8038e+00,\n",
       "             4.0820e+00,  1.4626e+01],\n",
       "           [ 7.9062e+00, -6.7142e+00,  1.7787e+01,  ...,  3.7473e+01,\n",
       "             3.3630e+01, -9.7041e+00]]],\n",
       "\n",
       "\n",
       "         [[[-1.4467e+01,  1.1045e+01,  1.6891e+01,  ..., -1.4389e+01,\n",
       "            -1.9689e+01, -1.2884e+00],\n",
       "           [-5.6647e+01,  3.1731e+01,  4.1175e+01,  ..., -1.3485e+01,\n",
       "             1.0682e+00, -2.8401e+00],\n",
       "           [-5.1114e+01, -3.2714e+01,  5.7726e-01,  ...,  4.6666e+01,\n",
       "             6.6659e+00, -2.9600e+01],\n",
       "           ...,\n",
       "           [-4.5523e+01,  2.1408e+00,  1.9916e+01,  ..., -1.8349e+01,\n",
       "             7.3188e+00,  2.0002e+01],\n",
       "           [-8.3082e+00, -3.7654e+00, -4.4376e+01,  ..., -8.8463e+00,\n",
       "             8.0850e+00, -2.4792e+01],\n",
       "           [ 2.5365e+01,  2.8706e+01,  8.4221e+00,  ...,  1.9864e+01,\n",
       "             2.6406e+01,  1.1368e+01]],\n",
       "\n",
       "          [[-1.2234e+01,  4.0134e+01,  4.3379e+01,  ..., -1.7311e+00,\n",
       "             7.3102e-01, -3.2872e+01],\n",
       "           [ 2.5139e+01,  7.7825e+00, -3.3351e+01,  ..., -3.2151e+01,\n",
       "            -5.4775e+00,  1.4894e+01],\n",
       "           [-4.1758e+01,  1.4833e+01,  4.3958e+01,  ...,  2.6399e+01,\n",
       "             8.9779e+00,  5.3559e+00],\n",
       "           ...,\n",
       "           [ 1.4629e+01,  3.9425e+00, -1.9638e+01,  ..., -3.9936e+01,\n",
       "             5.0196e+00, -3.2517e+01],\n",
       "           [-1.8632e+01, -3.4066e+01, -3.5524e+00,  ..., -2.9207e+01,\n",
       "             1.5795e+01,  2.8242e+01],\n",
       "           [-1.1732e+01, -9.5148e+00, -4.9238e+01,  ...,  1.6775e+00,\n",
       "             2.3951e+00,  1.8427e+01]],\n",
       "\n",
       "          [[-2.1467e-01, -4.0516e+01, -2.5352e+01,  ..., -7.8434e+00,\n",
       "            -4.9252e-01,  3.7592e+00],\n",
       "           [ 2.3220e+01,  1.2871e+01, -1.1955e+01,  ..., -1.2693e+01,\n",
       "            -3.6629e+01, -5.0260e+01],\n",
       "           [ 3.3978e+01,  1.3507e+01, -1.0176e+01,  ...,  1.8435e+01,\n",
       "             1.6312e+01, -1.1633e+00],\n",
       "           ...,\n",
       "           [-1.9155e+01, -2.6400e+01, -5.2510e+00,  ...,  3.7622e+00,\n",
       "             9.6443e+00, -1.4957e+01],\n",
       "           [-1.8356e+01,  4.9431e-02,  9.8868e+00,  ...,  3.9096e+01,\n",
       "             9.1545e+00, -2.9414e+00],\n",
       "           [ 2.9942e+00, -3.8068e+00,  1.0815e+01,  ..., -2.5064e+01,\n",
       "            -1.8968e+01, -4.2721e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 3.1365e+01,  1.7389e+01,  7.6809e+00,  ...,  5.8277e+00,\n",
       "             1.1406e+01, -1.1616e+01],\n",
       "           [ 2.0080e+01,  1.9793e+01, -7.2927e+00,  ..., -1.3225e+01,\n",
       "             2.2623e+01, -1.3791e+01],\n",
       "           [ 1.0539e+01, -6.9095e+00,  3.0719e+01,  ...,  1.8580e+01,\n",
       "            -1.2244e+01, -1.5796e+01],\n",
       "           ...,\n",
       "           [ 2.4218e+01,  4.9020e-01, -1.8969e+01,  ...,  5.9670e+00,\n",
       "             8.4511e+00,  3.9851e+01],\n",
       "           [-2.0268e+01, -4.0079e+01, -1.5422e+01,  ..., -5.0912e+01,\n",
       "            -3.0448e+01,  1.3084e+00],\n",
       "           [-2.9720e+00, -1.4111e+00, -1.5542e+01,  ..., -2.2438e+01,\n",
       "            -3.3941e+01, -3.1782e+01]],\n",
       "\n",
       "          [[-3.0159e+01, -1.1415e+01, -2.0270e+01,  ...,  2.6391e+01,\n",
       "             1.8357e+01,  1.0301e+01],\n",
       "           [ 1.9444e+01,  4.0299e+01,  1.5172e+01,  ...,  2.8983e+00,\n",
       "            -2.7186e+00, -2.9226e+01],\n",
       "           [ 2.0373e+01,  2.3773e+01,  1.3727e+01,  ...,  2.0495e+01,\n",
       "             1.4317e+01,  1.6943e+01],\n",
       "           ...,\n",
       "           [-1.0922e+01,  1.9295e+01,  1.0484e+01,  ...,  4.8413e+00,\n",
       "             1.0446e+01,  1.1701e+01],\n",
       "           [-1.5306e+01,  2.6730e+01, -1.9884e+01,  ..., -3.9152e+00,\n",
       "            -9.6882e+00,  1.4332e+01],\n",
       "           [-2.8785e+01,  7.8686e+00,  5.7626e+01,  ..., -4.0656e+01,\n",
       "             7.0854e+00, -8.2252e+00]],\n",
       "\n",
       "          [[-1.2634e+01,  1.3315e+01, -2.6106e+01,  ...,  3.8735e+01,\n",
       "            -6.8185e+00, -3.3109e+01],\n",
       "           [-3.4574e-01,  2.9427e+01, -2.4499e+01,  ...,  1.0566e+01,\n",
       "            -1.5086e+01,  2.2931e+01],\n",
       "           [ 2.4027e+01, -1.9753e+01,  4.5199e+01,  ...,  2.3163e+01,\n",
       "            -1.0345e+01, -3.1916e+01],\n",
       "           ...,\n",
       "           [-3.8081e+01, -1.3473e+01, -7.0790e+00,  ..., -8.3745e+00,\n",
       "             2.4611e+01,  1.2799e+01],\n",
       "           [ 4.7216e-01,  2.6853e+01, -1.3627e+01,  ...,  3.2651e+00,\n",
       "            -4.0965e+00,  1.5869e+01],\n",
       "           [ 8.7585e+00,  2.8264e+01, -2.7273e+00,  ..., -2.9075e+00,\n",
       "             7.0081e-01, -5.8993e+00]]],\n",
       "\n",
       "\n",
       "         [[[ 4.3195e+00, -3.4216e+01, -2.0338e+01,  ..., -6.7931e+00,\n",
       "             1.1332e+01, -3.2315e+01],\n",
       "           [ 1.7496e+01, -5.8809e+00,  1.6602e+01,  ...,  1.6658e+01,\n",
       "             8.2339e+00,  4.2695e+01],\n",
       "           [-1.9347e+01, -1.6690e+01, -1.8149e+01,  ..., -4.2775e+00,\n",
       "            -2.4955e+00,  3.1832e+00],\n",
       "           ...,\n",
       "           [ 9.1060e+00,  1.2897e+01,  2.4209e+01,  ...,  1.2200e+01,\n",
       "             2.0761e+01, -1.9529e+01],\n",
       "           [-2.4488e+01, -4.2731e+00,  9.5064e+00,  ..., -1.3368e+01,\n",
       "             2.3857e+01,  2.3651e+01],\n",
       "           [ 2.3818e+01,  2.1009e+00,  2.3942e+01,  ...,  2.7535e+01,\n",
       "             4.1507e+01,  1.6287e+01]],\n",
       "\n",
       "          [[ 1.7309e+00,  2.3906e+01,  4.5787e+01,  ...,  2.1579e+01,\n",
       "            -1.2364e+01, -2.5820e+01],\n",
       "           [-1.7501e+01, -3.3948e+01,  1.2763e+00,  ...,  5.1091e+01,\n",
       "             2.4358e+01,  4.4033e+01],\n",
       "           [-3.0211e+01, -1.8210e+00, -1.9514e+01,  ...,  4.4400e+00,\n",
       "            -8.7703e+00, -5.8971e+00],\n",
       "           ...,\n",
       "           [ 1.9514e+01, -2.4097e+01,  4.0811e+00,  ...,  3.0532e+01,\n",
       "            -3.7247e+00,  1.8268e+01],\n",
       "           [-4.3840e+00, -3.2532e+00,  4.1344e+00,  ..., -5.3034e+01,\n",
       "            -4.5204e+00, -1.3201e+01],\n",
       "           [-4.8904e+01,  7.6154e+00,  3.1368e+01,  ..., -1.3802e+01,\n",
       "             3.6313e+01, -7.4160e-01]],\n",
       "\n",
       "          [[ 2.0251e+01, -2.0710e+01,  3.0744e+01,  ...,  1.3271e+00,\n",
       "            -1.4701e+01,  1.1246e+01],\n",
       "           [ 1.4035e+01,  8.6536e+00,  1.7043e+01,  ...,  8.3049e-01,\n",
       "             2.0661e+01,  2.5646e+01],\n",
       "           [-2.7299e+01,  1.1606e+01, -1.7220e+01,  ...,  2.5219e+01,\n",
       "             1.1512e+01, -3.1345e+00],\n",
       "           ...,\n",
       "           [ 8.8612e+00, -1.9914e+01, -1.1084e+01,  ..., -3.0497e+01,\n",
       "             2.3523e+01,  4.6283e+01],\n",
       "           [ 1.6235e+00, -2.2754e+01,  2.8052e+01,  ..., -2.9892e+01,\n",
       "            -4.3535e-01,  9.1374e+00],\n",
       "           [ 1.0505e+01, -5.8678e+00, -2.3175e+01,  ..., -2.5505e+01,\n",
       "             2.0614e+01, -7.6092e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 9.0695e+00, -7.1525e+00,  2.3679e+01,  ..., -1.1682e+01,\n",
       "            -1.8653e+01,  4.3591e+01],\n",
       "           [ 3.4279e+00, -1.1450e+01,  6.6482e+00,  ..., -7.8140e-01,\n",
       "             3.4348e+01,  9.6711e+00],\n",
       "           [-8.7309e+00, -2.8155e+00, -2.9585e+01,  ...,  6.3369e+00,\n",
       "             1.4223e+01,  1.2964e+00],\n",
       "           ...,\n",
       "           [ 9.2164e+00,  5.3540e+00,  2.3005e+01,  ..., -4.7338e+00,\n",
       "             5.2182e+01, -1.0019e+01],\n",
       "           [-2.9722e+01, -9.8014e+00, -3.0587e+01,  ...,  4.7672e+00,\n",
       "            -8.2530e+00, -2.5665e+01],\n",
       "           [ 1.0559e+01,  1.1991e+01,  3.6719e+01,  ...,  4.3299e+01,\n",
       "            -1.8727e+00,  4.8050e+01]],\n",
       "\n",
       "          [[ 4.1215e+01,  1.5944e+01,  2.6480e+01,  ...,  2.8550e+00,\n",
       "             6.3433e+00,  7.9703e+00],\n",
       "           [ 9.8590e+00, -2.7258e+00,  3.8134e+01,  ...,  5.5420e-01,\n",
       "            -3.1577e+01, -2.0454e+01],\n",
       "           [ 3.6157e-01, -4.4832e+00, -3.1394e+01,  ..., -4.4025e+00,\n",
       "            -1.0902e+00, -1.2102e+01],\n",
       "           ...,\n",
       "           [ 1.2902e+01,  4.5156e+00, -2.0926e+00,  ...,  3.6229e+01,\n",
       "            -9.2273e-01,  1.2739e+01],\n",
       "           [-1.2803e+01, -3.6828e+01, -2.3351e+01,  ..., -4.2524e+00,\n",
       "             1.0345e+01, -2.3898e+01],\n",
       "           [-8.5283e+00, -4.2383e+01, -7.4328e+00,  ..., -1.0490e+01,\n",
       "            -4.2905e+01, -2.1271e+01]],\n",
       "\n",
       "          [[-1.6204e+01,  1.1954e+01,  1.9976e+01,  ..., -1.6165e+01,\n",
       "            -1.2254e+01,  1.8313e+01],\n",
       "           [-1.0997e+00, -4.0497e+00,  4.9303e+00,  ...,  7.3441e+00,\n",
       "             2.8412e+01,  3.2919e+01],\n",
       "           [ 4.5987e+01,  1.5409e+01,  2.1201e+01,  ...,  1.6856e+01,\n",
       "            -1.1830e+01,  1.0399e+01],\n",
       "           ...,\n",
       "           [-2.2664e+01,  1.1989e+01, -1.3193e+01,  ..., -9.2548e+00,\n",
       "            -1.6042e+01,  8.3416e+00],\n",
       "           [-1.4229e+01,  1.3553e+01,  8.1281e-02,  ...,  3.2247e+01,\n",
       "            -3.1672e+01,  1.4757e+01],\n",
       "           [-1.3311e+01,  2.3198e+01, -1.1765e+01,  ..., -3.2740e+01,\n",
       "            -1.9541e+01,  1.9642e+01]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[ 3.4242e+00,  2.1507e+01, -1.6101e+00,  ...,  1.4300e+01,\n",
       "            -1.2378e+01, -2.5162e+01],\n",
       "           [ 1.0386e-01,  7.4904e+00, -2.0476e+01,  ...,  1.7761e+01,\n",
       "             1.4743e+01, -1.0568e+00],\n",
       "           [-7.9076e+00, -5.9890e+00,  5.7669e+00,  ..., -5.4621e+00,\n",
       "             1.2321e+01, -2.1836e+01],\n",
       "           ...,\n",
       "           [-1.3338e+01,  4.2418e+00,  1.2456e+01,  ...,  1.1030e+01,\n",
       "             2.4221e+01,  5.0304e+00],\n",
       "           [-1.6263e+01, -1.0451e+01,  3.9246e+01,  ..., -2.2591e+00,\n",
       "             2.8346e+01,  1.0398e+01],\n",
       "           [-1.5461e+01, -3.0019e+01,  1.5846e+01,  ..., -5.2009e+00,\n",
       "            -2.6020e+01, -2.4331e+01]],\n",
       "\n",
       "          [[-2.6637e+01, -6.7110e+00, -4.8629e+01,  ...,  1.8015e+01,\n",
       "             4.3023e-04,  5.1152e+01],\n",
       "           [-2.2332e+01,  2.3884e+01,  3.9932e+00,  ...,  2.9755e+01,\n",
       "             2.2463e+01, -3.4621e+01],\n",
       "           [ 8.4776e+00,  3.4361e+01,  1.6132e+01,  ..., -9.5897e+00,\n",
       "            -3.2776e+01, -5.3557e+00],\n",
       "           ...,\n",
       "           [ 2.4022e+00, -1.8002e+01,  7.5503e+00,  ..., -1.5283e+01,\n",
       "            -1.5195e+01,  9.5091e+00],\n",
       "           [-1.1361e+01,  3.0906e+01, -6.8273e+00,  ..., -2.7622e+00,\n",
       "            -2.4781e+01,  1.1471e+01],\n",
       "           [-2.5736e+00, -4.7803e+00, -2.7397e+00,  ..., -3.4314e+00,\n",
       "             7.1628e+00,  1.4397e+01]],\n",
       "\n",
       "          [[ 1.2956e+01, -3.6763e+00,  2.1927e+01,  ..., -8.7025e+00,\n",
       "             1.8573e+01, -2.2153e+01],\n",
       "           [-5.2937e+00,  4.2938e+01, -2.7858e+01,  ...,  1.9036e+01,\n",
       "             2.8433e-01,  3.7989e+00],\n",
       "           [-2.3008e+01, -1.5195e+01,  8.9310e+00,  ..., -1.7472e+00,\n",
       "             3.0438e+01,  1.9624e+01],\n",
       "           ...,\n",
       "           [-2.1061e+01,  1.1679e+01, -3.1753e+01,  ...,  7.6200e+00,\n",
       "             1.6595e+01, -2.4975e+01],\n",
       "           [-6.0852e+00, -3.1232e+01, -1.0087e+01,  ...,  9.7207e+00,\n",
       "            -1.9614e+01, -9.7011e+00],\n",
       "           [-5.9972e+00,  1.8419e+01,  1.4936e+01,  ...,  5.5531e+01,\n",
       "            -7.5461e+00, -8.4814e-01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.1704e+01,  7.4959e-01,  1.2348e+01,  ..., -3.1478e+01,\n",
       "             1.2335e+01,  1.2558e+01],\n",
       "           [-2.9742e+00,  1.4096e+01, -1.2131e+01,  ...,  5.9820e+00,\n",
       "             5.0089e+00,  9.0898e+00],\n",
       "           [-2.3461e+00, -2.2309e+01, -2.8195e+01,  ...,  1.2588e+01,\n",
       "            -2.7241e+01,  1.7808e+01],\n",
       "           ...,\n",
       "           [ 6.7416e+01,  1.9553e+01, -1.5392e+01,  ...,  6.7781e+01,\n",
       "             1.3779e+01,  4.8799e+01],\n",
       "           [ 2.4050e+01,  1.6921e+01, -1.7813e+00,  ..., -1.7744e+01,\n",
       "             4.4301e+00, -1.0183e+01],\n",
       "           [ 8.0169e+00,  3.5118e+00, -5.8632e+00,  ...,  6.9704e+00,\n",
       "             6.3500e+00,  2.5209e+01]],\n",
       "\n",
       "          [[-1.9289e+01,  1.6219e+00,  1.6784e+01,  ...,  1.4503e+01,\n",
       "             8.1319e+00,  1.9794e+00],\n",
       "           [-3.0188e+01,  2.4947e+01,  3.2406e+01,  ...,  2.0779e+01,\n",
       "            -1.3464e+01,  2.9952e+01],\n",
       "           [ 1.7905e+01, -2.9379e+00,  2.9384e+01,  ...,  6.4629e+00,\n",
       "             3.3160e+01,  1.0170e+01],\n",
       "           ...,\n",
       "           [ 2.3463e+01,  5.7400e-01, -2.3486e+01,  ..., -2.5411e+01,\n",
       "             1.6788e+01,  2.4591e+01],\n",
       "           [ 1.4960e+01,  4.3584e+01,  1.6196e+01,  ...,  1.0953e+01,\n",
       "            -5.7509e+00,  1.3278e+01],\n",
       "           [-1.6767e+01, -4.8797e+00,  2.3617e+01,  ..., -2.6720e+00,\n",
       "            -2.0222e+01, -1.3158e+00]],\n",
       "\n",
       "          [[-1.6773e+01,  2.7350e+01,  7.3937e+00,  ..., -1.7937e+01,\n",
       "             2.1426e+00, -1.0863e+01],\n",
       "           [ 1.1841e+00, -1.4582e+01, -8.4813e+00,  ...,  3.0907e+00,\n",
       "             9.3721e+00,  2.4333e+01],\n",
       "           [-5.1390e+01, -1.6293e+01, -7.5266e+00,  ..., -1.3565e+01,\n",
       "             3.2661e+00,  6.9754e+00],\n",
       "           ...,\n",
       "           [-1.5535e+01,  2.3288e+01,  7.2033e+00,  ...,  3.8503e+00,\n",
       "             1.9399e+01,  1.5826e+01],\n",
       "           [ 1.6652e+01, -9.4666e+00, -2.4234e+01,  ..., -5.9071e+00,\n",
       "             1.3105e+01, -2.1589e+01],\n",
       "           [ 9.3584e+00, -1.4625e+01, -3.3828e+01,  ...,  1.7888e+01,\n",
       "            -1.5335e+01,  3.6294e+01]]],\n",
       "\n",
       "\n",
       "         [[[-1.2983e+01,  1.9621e+01,  6.8632e+00,  ...,  4.4238e-01,\n",
       "            -3.9052e+00,  1.9106e+01],\n",
       "           [-5.7050e+00,  1.3335e+01,  3.1604e-01,  ..., -3.8002e+00,\n",
       "             9.8004e+00, -1.1043e+01],\n",
       "           [-9.2924e+00,  1.8135e+01,  2.4789e+00,  ...,  4.6715e-01,\n",
       "             8.9180e+00, -4.5190e+00],\n",
       "           ...,\n",
       "           [-3.6034e+01,  3.7304e+01, -3.6723e+01,  ..., -8.9463e+00,\n",
       "            -1.6335e+01,  2.6297e-01],\n",
       "           [-4.8129e+01, -1.0164e+01,  8.1125e-02,  ...,  1.9229e+01,\n",
       "            -2.8685e+01, -1.5962e+01],\n",
       "           [ 3.2313e+01, -5.2122e+00,  1.6796e+00,  ..., -1.9822e+01,\n",
       "            -1.3054e+01,  5.2529e+01]],\n",
       "\n",
       "          [[-2.5266e+01, -5.5602e+01, -3.9236e-01,  ...,  4.7035e+00,\n",
       "             1.6361e+00, -3.7101e+00],\n",
       "           [-2.2403e+01, -1.9403e+01, -2.0498e+01,  ...,  8.9105e+00,\n",
       "            -3.0524e+01, -2.8996e+00],\n",
       "           [ 2.1123e+01, -1.6764e+00, -2.6443e+01,  ...,  1.2149e+01,\n",
       "             1.9526e+01, -9.0601e+00],\n",
       "           ...,\n",
       "           [-1.2864e+01, -1.5381e+01,  3.8505e+01,  ...,  1.6866e+01,\n",
       "            -4.6884e-01, -4.8540e-01],\n",
       "           [-1.4534e+01,  3.5775e+00, -6.6504e+00,  ...,  9.0301e+00,\n",
       "            -1.1282e+01,  8.4849e-01],\n",
       "           [-1.1394e+01, -4.2795e-01,  7.1145e+00,  ..., -3.0948e+01,\n",
       "            -1.1379e+01, -1.7802e+01]],\n",
       "\n",
       "          [[ 2.6553e+01,  2.1415e-01, -3.3455e+01,  ..., -1.9824e+01,\n",
       "            -1.2641e+01,  6.8998e+00],\n",
       "           [ 1.6601e+00,  1.9221e+01,  5.9481e+00,  ...,  2.7419e+00,\n",
       "            -6.4202e+00, -1.7621e+01],\n",
       "           [ 1.7179e+01, -1.2923e+01,  4.1977e+00,  ..., -3.5820e+00,\n",
       "            -5.9288e+00, -4.1626e+01],\n",
       "           ...,\n",
       "           [-1.2577e+01,  2.7344e+00,  9.8828e+00,  ..., -5.2039e+01,\n",
       "            -1.6699e+01,  2.5260e+01],\n",
       "           [-1.2066e+00, -1.9130e+01,  8.9702e+00,  ...,  2.6205e+01,\n",
       "             3.5715e+01,  1.3361e+01],\n",
       "           [ 1.8919e+01, -2.8329e+01,  3.0557e+00,  ..., -2.9733e+01,\n",
       "             1.1272e+01,  1.8505e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 2.2942e+01,  2.8011e+01, -1.1158e+01,  ...,  1.6622e+01,\n",
       "             1.4590e+01,  2.7996e+01],\n",
       "           [ 2.2092e+00, -1.7249e+01, -8.5264e+00,  ...,  2.0157e+01,\n",
       "             7.0455e+00,  1.9588e+01],\n",
       "           [-4.3382e+00,  1.2879e+01,  5.0921e+00,  ...,  2.7980e+00,\n",
       "             5.1086e+01,  1.1297e+01],\n",
       "           ...,\n",
       "           [ 3.8691e+01,  4.1055e+01,  1.3616e+01,  ...,  4.1515e+00,\n",
       "            -1.0567e-01, -6.1874e+00],\n",
       "           [ 1.3195e+01, -1.6732e+01,  3.1743e+01,  ..., -3.8641e+00,\n",
       "            -2.5390e+00,  1.3962e+01],\n",
       "           [ 4.5576e+01,  2.4784e+01, -1.2427e+01,  ..., -1.4464e+01,\n",
       "             9.8320e+00,  2.0645e+01]],\n",
       "\n",
       "          [[ 1.5763e+01, -2.0263e+01,  6.8173e+00,  ...,  7.5462e+00,\n",
       "             1.2826e+01,  3.1555e+01],\n",
       "           [-6.3526e+00,  2.1665e+01, -9.2670e+00,  ..., -1.5244e+01,\n",
       "            -1.5308e+01,  2.1403e+01],\n",
       "           [ 1.9688e+01, -1.1463e+01, -2.5559e+01,  ...,  1.4151e+01,\n",
       "             1.5540e+00,  4.5198e+00],\n",
       "           ...,\n",
       "           [ 8.3242e+00, -4.1917e+00, -4.7284e+01,  ...,  4.3256e+01,\n",
       "            -2.9966e+01,  3.9763e+01],\n",
       "           [ 3.4628e+01,  1.3392e+01, -2.1950e+01,  ..., -6.2767e+00,\n",
       "             3.1270e+01,  1.0700e+01],\n",
       "           [ 1.8757e+01,  5.7719e+00, -1.2767e+01,  ...,  1.4256e+01,\n",
       "            -3.2042e+01, -4.1422e+00]],\n",
       "\n",
       "          [[ 7.3506e+00, -1.2769e+01, -1.9809e+01,  ..., -2.4898e+01,\n",
       "            -3.8637e+00, -3.5839e+01],\n",
       "           [-3.2121e+00,  8.6094e+00,  3.0012e+00,  ..., -2.4240e+01,\n",
       "             1.3919e+01,  6.6757e+00],\n",
       "           [-4.6379e+00, -1.4516e+01,  1.6955e+01,  ..., -2.9008e+01,\n",
       "             1.4326e+01,  7.9492e-01],\n",
       "           ...,\n",
       "           [-3.8684e-01, -1.1520e+01, -1.5975e+00,  ..., -6.1567e+00,\n",
       "             3.8844e+00,  4.3272e+01],\n",
       "           [-1.3339e+01, -3.9807e-01, -4.2584e+00,  ...,  8.7122e+00,\n",
       "            -2.0287e+01, -3.8602e+01],\n",
       "           [ 1.9992e+01, -2.9965e+01,  2.1438e+01,  ..., -6.1224e+00,\n",
       "            -5.4034e+01,  1.8164e+01]]],\n",
       "\n",
       "\n",
       "         [[[-1.8342e+01, -7.3876e+00, -9.3287e+00,  ..., -1.0955e+01,\n",
       "            -1.2515e+01, -5.0580e+01],\n",
       "           [ 8.8906e+00, -1.0658e+01,  3.4271e+01,  ...,  1.5124e+00,\n",
       "             9.1995e+00,  3.2102e+00],\n",
       "           [-1.8411e+01,  1.2567e+01, -2.6494e+01,  ...,  3.9834e+00,\n",
       "             8.2613e+00,  1.4112e+01],\n",
       "           ...,\n",
       "           [ 3.0889e+01,  2.0157e+01,  3.2835e+01,  ..., -2.7089e+01,\n",
       "            -5.7085e+00, -1.3512e+01],\n",
       "           [ 3.8616e+00,  2.5666e+00, -5.7146e+00,  ...,  2.2987e+01,\n",
       "             6.3909e+00, -1.8027e+00],\n",
       "           [-1.0265e+01, -5.2774e+00, -2.0598e+01,  ..., -4.7356e+01,\n",
       "             2.2658e+01, -4.2725e+00]],\n",
       "\n",
       "          [[-2.0661e+01, -5.9817e+00, -1.2839e+01,  ...,  2.8422e+01,\n",
       "            -3.1542e+01,  1.6098e+01],\n",
       "           [-2.4961e-01, -3.8912e+00,  2.3705e+01,  ...,  3.0605e+00,\n",
       "             5.8440e+00,  4.8671e+01],\n",
       "           [ 1.4165e+01, -2.7204e+01,  1.5489e+01,  ...,  1.3807e+00,\n",
       "            -2.7550e+01, -4.7413e+00],\n",
       "           ...,\n",
       "           [ 4.2762e+01,  1.2309e+00,  2.9576e+00,  ...,  2.1675e+01,\n",
       "            -3.3126e+01,  1.1588e+01],\n",
       "           [ 1.9199e+01,  2.1178e+01,  9.3561e-01,  ...,  1.2657e+01,\n",
       "            -2.4284e+01,  1.1057e+01],\n",
       "           [-1.8630e+01,  8.8345e+00,  9.3950e-01,  ...,  3.2137e+01,\n",
       "            -2.8180e+01, -3.2820e+01]],\n",
       "\n",
       "          [[ 2.6131e+01, -1.5117e+01,  5.8514e+00,  ...,  4.7136e+00,\n",
       "            -7.8318e+01,  2.8487e-01],\n",
       "           [ 5.2481e+00, -7.9333e+00,  2.5853e+01,  ...,  2.0952e+01,\n",
       "             2.2421e+01,  5.1936e+00],\n",
       "           [-1.8755e+01,  3.0568e+00, -5.4375e+00,  ..., -7.1730e+00,\n",
       "             1.7272e+01,  1.8286e+00],\n",
       "           ...,\n",
       "           [-3.1690e+00, -6.5224e+00, -4.8952e+00,  ...,  9.5396e+00,\n",
       "            -1.1189e+01,  1.2191e+01],\n",
       "           [ 5.2778e+01, -1.7822e+01, -9.3712e+00,  ...,  6.1109e+00,\n",
       "             1.9366e+01,  2.8251e+01],\n",
       "           [-3.0199e+01,  2.9036e+01,  2.7914e+00,  ...,  3.3225e+01,\n",
       "             3.4434e+01,  5.6872e+00]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-2.5133e+01,  2.0804e+01,  9.9494e+00,  ...,  6.4851e+00,\n",
       "            -3.7583e+01, -3.1231e+01],\n",
       "           [ 6.2202e+01, -1.5636e+01, -1.2535e+01,  ..., -3.2150e+01,\n",
       "             2.2354e+01, -7.7817e+00],\n",
       "           [-2.4099e+01, -2.8766e+01,  1.9315e+01,  ..., -8.5004e+00,\n",
       "             1.0259e+00,  1.8644e+01],\n",
       "           ...,\n",
       "           [-3.2282e+01, -1.9000e+01, -2.0644e+01,  ...,  3.4773e+00,\n",
       "             2.8946e+00, -1.6095e+01],\n",
       "           [-6.9757e+00,  2.6070e+01, -1.2848e+01,  ...,  2.6483e+01,\n",
       "            -2.4616e+00,  1.9850e+01],\n",
       "           [ 2.6984e+01,  2.5431e+00,  1.1125e+01,  ...,  1.1947e+01,\n",
       "            -3.0661e+00, -2.1254e+01]],\n",
       "\n",
       "          [[ 2.5945e+01,  6.9706e-01,  6.9011e+00,  ...,  1.6665e+01,\n",
       "            -1.4153e+01, -7.4751e+00],\n",
       "           [ 1.1162e+01,  2.0029e+01,  1.5079e+01,  ..., -2.7780e+00,\n",
       "            -1.5977e+01, -2.0093e+01],\n",
       "           [ 9.6512e+00, -3.3335e+01,  1.0517e+01,  ..., -4.7202e+01,\n",
       "            -2.3810e+01, -1.0963e+01],\n",
       "           ...,\n",
       "           [-6.4345e+00, -3.3273e+01, -2.8779e+01,  ...,  7.4600e+00,\n",
       "            -3.1756e+01,  4.9637e+00],\n",
       "           [-2.1490e+01, -4.6437e+01,  4.9790e+00,  ...,  8.9711e+00,\n",
       "             2.9468e+01,  1.0440e+01],\n",
       "           [ 3.3223e+01,  4.2474e+01, -2.8785e+01,  ...,  4.0769e+01,\n",
       "            -3.1284e+01,  2.2796e+01]],\n",
       "\n",
       "          [[-1.8131e+01, -2.8002e+01,  1.4826e+01,  ..., -2.2273e+01,\n",
       "            -1.3793e+01,  2.0629e+01],\n",
       "           [ 2.8457e+01,  2.0295e+01, -5.5432e+00,  ..., -1.3775e+01,\n",
       "             1.7216e+01,  3.5054e+01],\n",
       "           [ 1.8136e+00,  2.1810e+00, -3.4371e+01,  ...,  6.4043e+00,\n",
       "             3.3198e+01, -1.1829e+01],\n",
       "           ...,\n",
       "           [ 1.9092e+01,  3.4708e+01, -1.2422e+01,  ..., -1.1817e+01,\n",
       "            -6.5987e+00, -1.0068e+01],\n",
       "           [ 3.5537e+01,  1.0849e+01, -1.8203e+00,  ..., -7.5761e+00,\n",
       "            -2.8293e+01,  1.8400e+01],\n",
       "           [ 2.7749e+00, -4.1052e+01,  3.0818e+01,  ..., -1.8542e+01,\n",
       "            -3.0376e+01,  1.3491e+01]]]],\n",
       "\n",
       "\n",
       "\n",
       "        [[[[-2.4978e+01, -2.9873e+01, -2.8712e+01,  ...,  4.7791e+00,\n",
       "            -2.7035e+01,  3.4054e+01],\n",
       "           [-4.2461e+01,  2.8528e+01, -9.6076e+00,  ..., -1.6849e+01,\n",
       "            -2.1634e+01,  6.4732e+00],\n",
       "           [-1.9456e+01,  2.3356e+01, -1.9899e+01,  ..., -1.1136e+01,\n",
       "             3.5097e+01,  2.3671e+01],\n",
       "           ...,\n",
       "           [ 2.4293e+01, -2.8423e+01,  1.5603e+00,  ...,  1.9158e+01,\n",
       "             3.2131e+00,  5.6812e+01],\n",
       "           [ 2.9999e+01, -1.4975e+01, -1.5311e-01,  ...,  2.9325e+01,\n",
       "            -7.3780e+00, -1.2193e+01],\n",
       "           [-7.6551e-03, -1.5147e+01,  4.4260e+00,  ...,  9.0752e+00,\n",
       "             1.9602e+01, -2.1953e+01]],\n",
       "\n",
       "          [[ 1.4285e+01,  8.8183e+00, -1.8729e+01,  ..., -2.1126e+00,\n",
       "             9.3692e+00, -1.4092e+01],\n",
       "           [ 1.5228e+01, -7.7680e+00, -9.0628e+00,  ...,  1.6540e+01,\n",
       "             7.2181e+00,  5.3043e+00],\n",
       "           [ 2.6341e+00,  3.0012e+01,  3.6132e+01,  ..., -5.6441e+00,\n",
       "            -9.1808e+00,  8.4580e+00],\n",
       "           ...,\n",
       "           [-2.1348e+01,  1.1937e+01,  1.1548e+01,  ..., -7.6446e+00,\n",
       "            -2.7358e+01,  6.1525e+00],\n",
       "           [-4.4618e-01, -1.8040e+01,  6.3560e+00,  ...,  3.2786e+01,\n",
       "            -2.5911e+01,  5.8878e+00],\n",
       "           [-6.6854e+00, -1.0304e+01,  1.3939e+01,  ..., -2.0999e+00,\n",
       "            -2.5492e+01, -3.2852e+01]],\n",
       "\n",
       "          [[ 7.6820e+00,  7.9782e+00,  9.5856e+00,  ...,  2.8318e+01,\n",
       "            -4.6973e+01, -1.2458e+01],\n",
       "           [ 1.1935e+01, -1.6859e+00,  6.5021e+00,  ..., -3.6287e+00,\n",
       "             1.3516e+01,  2.5460e+01],\n",
       "           [-1.5966e+01, -9.0939e+00, -1.1352e+01,  ...,  2.4584e+00,\n",
       "            -1.4626e+00, -1.3838e+01],\n",
       "           ...,\n",
       "           [ 2.8393e+01,  1.4861e+01, -8.4382e+00,  ...,  1.3935e+01,\n",
       "            -4.4009e+00,  2.2245e+00],\n",
       "           [-7.3546e-01,  1.8700e+01,  2.0246e+01,  ..., -3.8955e+00,\n",
       "            -2.2013e+00,  2.1702e+01],\n",
       "           [-1.4926e+01,  1.0793e+01, -8.3575e+00,  ..., -2.1054e+01,\n",
       "            -2.1764e+01,  1.8465e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[-4.5307e+00, -1.1527e+01, -1.4981e+01,  ...,  1.1323e+00,\n",
       "            -6.5151e+00,  1.8070e+01],\n",
       "           [ 1.4483e+01, -7.6899e+00, -5.3478e+01,  ..., -4.9177e+00,\n",
       "             6.8859e+01, -2.9061e+00],\n",
       "           [-1.1390e+01,  3.0244e+01, -3.9633e+01,  ..., -7.0341e+00,\n",
       "             1.2051e+01, -1.6466e+00],\n",
       "           ...,\n",
       "           [-1.5822e+01,  3.6386e+01, -2.0944e+01,  ..., -2.3374e+01,\n",
       "             2.0749e+00, -4.3062e-01],\n",
       "           [ 3.8171e+01,  1.4846e+01, -3.4202e+01,  ..., -2.3906e+00,\n",
       "             1.8011e+01, -3.4944e+01],\n",
       "           [-3.1253e+01, -1.5000e+01, -8.8255e+00,  ..., -2.1859e+01,\n",
       "            -4.9077e+01, -7.5191e+00]],\n",
       "\n",
       "          [[-1.1696e+01, -3.3615e+01,  2.7042e+01,  ..., -3.1214e+01,\n",
       "             4.9053e+00, -3.1331e+01],\n",
       "           [-6.4209e+01, -1.1015e+01,  1.2349e+01,  ..., -1.5569e+00,\n",
       "            -3.9704e+00, -8.7110e+00],\n",
       "           [ 8.3291e+00,  1.3189e+01, -1.7552e+01,  ..., -1.1454e+00,\n",
       "             3.2804e+01,  1.0863e+01],\n",
       "           ...,\n",
       "           [-3.0292e+01,  8.8841e+00, -7.4792e-01,  ...,  1.7561e+00,\n",
       "             3.8599e+01,  4.1382e+01],\n",
       "           [ 7.9252e-01, -1.1887e+01,  1.1855e+00,  ...,  4.4532e+01,\n",
       "             2.4121e+01, -1.4335e+01],\n",
       "           [ 3.5398e+01,  2.9918e+01,  2.6592e+01,  ...,  7.1897e+00,\n",
       "             3.4795e+01, -4.2115e+00]],\n",
       "\n",
       "          [[ 1.1904e+01,  4.4998e-01,  8.1613e+00,  ...,  5.1957e+00,\n",
       "            -1.4945e+01, -3.2744e+01],\n",
       "           [ 4.1308e+00, -1.2672e+01,  2.0105e+01,  ..., -6.6441e+00,\n",
       "             1.1997e+01,  2.7499e+01],\n",
       "           [ 3.3160e+01,  1.8447e+01, -5.8071e+00,  ..., -5.9023e+00,\n",
       "             9.1736e+00,  1.4119e+01],\n",
       "           ...,\n",
       "           [ 1.3641e+01,  3.2061e+00,  2.4664e+00,  ...,  4.6351e+00,\n",
       "             9.3403e+00, -5.7835e+00],\n",
       "           [-2.1044e+01,  3.9089e+00,  1.5035e+01,  ..., -9.0096e+00,\n",
       "             3.8838e+01,  2.8319e+01],\n",
       "           [-1.8236e+00, -2.5266e+00,  2.7958e+01,  ..., -2.4017e+00,\n",
       "             4.5459e+00,  1.6454e+01]]],\n",
       "\n",
       "\n",
       "         [[[-4.6547e+01,  1.9061e+00,  2.5274e+01,  ..., -2.6174e+01,\n",
       "             3.7046e+01,  2.8930e+01],\n",
       "           [ 7.2702e+00,  4.5549e+00, -9.4888e+00,  ..., -4.0569e+01,\n",
       "            -3.1913e+01,  1.9941e+00],\n",
       "           [-2.1464e+00, -1.2632e-01, -1.8905e+01,  ...,  5.5022e+00,\n",
       "            -5.5232e+01, -1.6277e+01],\n",
       "           ...,\n",
       "           [-1.9660e+00,  1.1004e+01, -8.1782e+00,  ...,  3.4581e+01,\n",
       "             8.2174e+00,  1.0841e+01],\n",
       "           [-3.1859e+01, -2.5331e+01, -8.3323e+00,  ...,  3.3196e+01,\n",
       "            -2.8497e+01, -2.6679e+00],\n",
       "           [ 6.6027e-01,  2.1370e+01,  1.4277e+00,  ...,  4.4860e+01,\n",
       "            -3.5649e+01,  1.4207e+01]],\n",
       "\n",
       "          [[-2.5715e+01, -6.9685e+00,  5.3232e+00,  ..., -2.2934e+01,\n",
       "             1.6901e+00, -6.2519e+00],\n",
       "           [ 2.0358e+01, -1.1081e+01, -3.2557e+01,  ..., -1.2257e+01,\n",
       "            -1.2849e+01,  1.4331e+01],\n",
       "           [-2.6879e+01, -1.6111e+00, -1.7846e+01,  ..., -4.6745e+00,\n",
       "             1.6516e+01,  1.4261e+01],\n",
       "           ...,\n",
       "           [-5.1620e+00, -1.5959e+01, -9.8469e+00,  ..., -3.1909e+01,\n",
       "            -1.0909e+01, -1.8166e+01],\n",
       "           [-3.1066e+00,  1.6991e+01, -9.0231e+00,  ..., -2.7443e+00,\n",
       "            -2.1202e+01, -3.0760e+01],\n",
       "           [ 1.5931e+01, -1.7148e+01, -1.3946e+00,  ...,  9.3864e+00,\n",
       "            -9.0466e+00, -1.2348e+01]],\n",
       "\n",
       "          [[-2.4751e+01,  1.4109e+01, -1.1266e+01,  ..., -3.0629e+01,\n",
       "            -2.8520e+01, -1.6644e+01],\n",
       "           [-6.3860e+00,  1.6591e+01, -1.2824e+01,  ...,  1.9416e+01,\n",
       "             2.7952e+01,  2.0792e+01],\n",
       "           [-8.7713e-01, -3.3808e+00, -1.6797e+01,  ..., -1.9986e+00,\n",
       "             1.4951e+01,  1.6507e+01],\n",
       "           ...,\n",
       "           [-1.6810e+01, -1.0768e+01,  1.4274e+01,  ...,  1.2552e+01,\n",
       "             1.9791e+01, -1.8739e+01],\n",
       "           [ 9.7439e+00,  5.6924e+00,  6.5841e+00,  ...,  5.8707e+00,\n",
       "            -1.3005e+01, -1.5732e+01],\n",
       "           [-4.5181e+01, -2.3493e+01, -1.1217e+01,  ...,  6.9589e+00,\n",
       "             1.3090e+01, -2.1795e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 8.4323e-01, -2.8810e+01, -1.8602e+01,  ..., -5.1184e+00,\n",
       "             3.9216e+01,  1.4794e+01],\n",
       "           [-3.8510e+00, -2.5290e+00,  4.7570e+01,  ..., -2.1772e+00,\n",
       "            -1.9843e+01, -1.1175e+01],\n",
       "           [ 1.0418e+01, -3.8470e+01,  6.2111e+00,  ...,  1.9355e+01,\n",
       "            -7.7431e+00,  1.3130e+01],\n",
       "           ...,\n",
       "           [-1.5352e+01, -5.7126e+01, -5.9485e+00,  ..., -3.2480e+01,\n",
       "            -4.0432e+01,  9.9793e+00],\n",
       "           [ 2.3614e+01,  4.4757e+01, -1.6626e+01,  ...,  1.2892e+01,\n",
       "             2.5748e+00, -2.2895e+01],\n",
       "           [ 1.0386e+01,  2.4769e-01,  2.0717e+01,  ...,  4.9651e+01,\n",
       "             8.9889e-01,  1.9297e+01]],\n",
       "\n",
       "          [[-1.8690e+01, -9.4277e+00, -1.4425e+01,  ..., -1.5693e+00,\n",
       "             1.2300e+01,  3.5172e+01],\n",
       "           [-1.9415e+01, -2.6860e+01, -1.0195e+01,  ..., -1.6789e+01,\n",
       "            -3.3202e+01, -3.8767e+01],\n",
       "           [-3.6613e+01,  6.5193e+00, -3.7128e+01,  ...,  2.3780e+01,\n",
       "             8.2330e+00, -1.7374e+01],\n",
       "           ...,\n",
       "           [ 9.6966e+00,  3.9114e+01,  6.8917e+00,  ...,  1.9543e+01,\n",
       "             4.3965e+00, -2.6239e+01],\n",
       "           [ 9.4270e+00,  7.9648e+00, -1.2331e+01,  ..., -1.3308e+01,\n",
       "            -1.0118e+01,  1.3269e+01],\n",
       "           [-1.0466e+01,  1.0287e+01, -7.9372e+00,  ..., -2.1908e-01,\n",
       "            -1.3123e+01,  8.0942e+00]],\n",
       "\n",
       "          [[-2.5102e+01, -2.0999e+01,  6.0210e+00,  ..., -2.8003e+01,\n",
       "             1.2315e+01, -1.3256e+01],\n",
       "           [-5.8154e+00,  1.4820e+01, -2.1722e+01,  ...,  3.1058e+01,\n",
       "             4.0939e+01, -2.1890e+01],\n",
       "           [ 8.7215e+00, -1.9579e+01, -1.7366e+01,  ...,  1.9813e+01,\n",
       "            -2.2119e+01, -1.5410e+00],\n",
       "           ...,\n",
       "           [ 1.0616e+01, -1.4472e+01, -3.1725e+01,  ..., -1.1307e+01,\n",
       "             3.6187e+00,  1.3228e+01],\n",
       "           [ 1.0365e+01, -1.0677e+00,  2.1023e+01,  ...,  3.7308e+01,\n",
       "             1.4679e-01, -2.8940e+00],\n",
       "           [ 1.6568e+00,  2.7718e+01, -2.5187e+01,  ..., -4.6138e+00,\n",
       "             3.3995e+01, -3.4954e+01]]],\n",
       "\n",
       "\n",
       "         [[[-2.0756e+01,  1.5194e+01, -5.4388e+01,  ...,  8.0852e+00,\n",
       "             2.7057e+01,  3.1017e-01],\n",
       "           [ 2.4100e+01, -1.7916e+01,  4.4932e+00,  ..., -3.1255e+01,\n",
       "            -1.3970e+00, -2.3302e+01],\n",
       "           [ 2.8748e+00, -4.2181e+00, -9.3648e-01,  ..., -3.9631e+00,\n",
       "            -2.4173e+01, -1.6001e+01],\n",
       "           ...,\n",
       "           [ 2.4779e+01,  1.9459e+01,  4.2873e+00,  ..., -9.9606e+00,\n",
       "            -7.6365e+00, -4.3531e+00],\n",
       "           [-7.3769e+00, -2.7617e+01,  1.2504e+01,  ...,  1.4634e+01,\n",
       "             2.5199e+00,  1.8449e+01],\n",
       "           [ 7.8790e+00,  9.3125e+00, -4.3100e+01,  ..., -2.2921e+01,\n",
       "            -6.6303e+01, -5.9917e+00]],\n",
       "\n",
       "          [[ 1.3448e+01, -3.4985e+01,  8.6729e+00,  ...,  1.2664e+01,\n",
       "             1.3249e+01, -2.0832e+01],\n",
       "           [-5.9098e+00, -2.4356e+01, -3.7125e+01,  ...,  2.3355e+01,\n",
       "            -1.1712e+01,  3.1060e+01],\n",
       "           [ 6.4689e+00, -1.6617e+01,  2.8634e+01,  ...,  2.6923e+01,\n",
       "             3.8884e-01,  1.6112e+01],\n",
       "           ...,\n",
       "           [-1.8696e+01,  8.5263e+00, -7.4251e+00,  ...,  2.1902e+01,\n",
       "             7.3364e+00,  6.9169e+00],\n",
       "           [ 8.1833e+00,  1.2288e+01,  5.9970e+01,  ..., -1.5782e+01,\n",
       "             2.5530e+01, -8.1755e+00],\n",
       "           [-1.0798e+01, -2.7472e+00, -3.1603e-01,  ...,  2.1199e+00,\n",
       "             1.8558e+01,  6.5743e+00]],\n",
       "\n",
       "          [[ 2.4821e+00, -2.6724e+01,  2.4344e+01,  ...,  4.3478e+00,\n",
       "            -1.8943e+01,  1.8741e+01],\n",
       "           [ 9.9997e+00,  4.6563e+00, -3.4202e+01,  ..., -2.8950e-02,\n",
       "             1.2512e+01, -1.2551e+01],\n",
       "           [-1.6415e+01,  1.7570e+00, -1.9353e+01,  ...,  9.2910e+00,\n",
       "             1.2510e+01,  3.5493e+00],\n",
       "           ...,\n",
       "           [-9.9588e+00, -4.5096e+01, -1.4700e+01,  ..., -2.7767e+01,\n",
       "             1.3381e+01,  3.4368e+01],\n",
       "           [-1.1595e+01, -1.5513e+01,  1.3333e+01,  ...,  1.3961e+01,\n",
       "            -1.1767e+01, -1.3911e+01],\n",
       "           [-3.7939e+01,  2.0481e+01,  1.2369e+01,  ...,  4.0805e+01,\n",
       "            -1.8567e+01,  2.7964e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 4.5780e+01, -1.9701e+01, -1.0706e+01,  ..., -2.2286e+01,\n",
       "            -2.4604e+01,  2.1450e+01],\n",
       "           [ 2.4041e+01,  4.2946e+00, -1.0709e+01,  ..., -1.8802e+01,\n",
       "            -1.3971e+00, -7.3565e+00],\n",
       "           [ 2.8410e+01,  8.4213e+00,  1.2736e+01,  ...,  2.3930e+01,\n",
       "             3.9858e+00,  1.6630e+01],\n",
       "           ...,\n",
       "           [ 1.6068e+00, -9.5432e+00,  2.4704e+01,  ...,  1.0969e+01,\n",
       "             2.5626e+01,  5.7191e+00],\n",
       "           [ 3.9530e+01,  2.3146e+01, -4.8055e+00,  ...,  4.7655e+00,\n",
       "            -6.1145e+00, -5.6862e+01],\n",
       "           [-2.8735e+01,  7.0884e+00,  1.0004e+01,  ...,  4.3298e+00,\n",
       "             1.7585e+01,  7.8828e+00]],\n",
       "\n",
       "          [[-1.6676e+01,  1.7719e+01, -3.8596e+01,  ..., -5.3040e+00,\n",
       "            -3.4352e+01,  1.5005e+01],\n",
       "           [-4.7553e+00, -1.4528e+01, -3.3012e+01,  ..., -1.4621e-01,\n",
       "            -1.5357e+01, -2.4575e+01],\n",
       "           [-3.0497e+00,  1.0221e+01,  2.0142e+01,  ..., -1.9905e+00,\n",
       "             2.4364e+01, -4.3716e+01],\n",
       "           ...,\n",
       "           [-1.4604e+01,  1.4018e+01, -1.4464e+01,  ...,  1.0433e+01,\n",
       "             3.8953e+00,  5.3195e+01],\n",
       "           [ 1.6094e+01, -1.7800e+01, -2.0304e+01,  ...,  1.5800e+00,\n",
       "            -1.7278e+01,  5.4756e+00],\n",
       "           [-7.7488e+00,  1.1335e+01,  2.5301e+01,  ...,  8.8663e+00,\n",
       "            -1.2593e+00,  9.0005e+00]],\n",
       "\n",
       "          [[-9.3289e+00, -4.4145e+00,  1.9769e+00,  ..., -2.3648e+01,\n",
       "            -1.6680e+01,  1.9732e+01],\n",
       "           [ 3.1026e+01, -1.4042e+01, -1.2745e+01,  ..., -2.4052e+01,\n",
       "             1.3137e+00,  2.2469e+01],\n",
       "           [ 4.6637e+00, -2.6330e+01, -9.4956e+00,  ...,  9.0409e+00,\n",
       "            -6.0156e+00, -2.7384e+01],\n",
       "           ...,\n",
       "           [-9.8520e-01,  1.8750e+01, -5.8236e+00,  ...,  1.4418e+01,\n",
       "             2.1480e+01,  1.2703e+01],\n",
       "           [ 2.4973e-01, -4.6969e+00,  2.9474e+00,  ..., -1.3365e+01,\n",
       "             1.1114e+01,  1.9510e+01],\n",
       "           [-1.8114e+01,  1.5375e+01,  9.5673e+00,  ..., -1.2578e+01,\n",
       "             9.7567e+00,  3.9479e+00]]],\n",
       "\n",
       "\n",
       "         ...,\n",
       "\n",
       "\n",
       "         [[[-9.0236e+00,  1.3529e+01, -1.2389e+01,  ..., -2.6655e+01,\n",
       "             1.1182e+01, -1.6799e+01],\n",
       "           [-1.5519e+01, -1.5510e+01, -2.6798e+01,  ...,  1.8154e-01,\n",
       "             9.6458e+00, -6.0717e+00],\n",
       "           [ 1.2466e+00, -2.5100e+01,  2.8655e+01,  ..., -5.9134e+00,\n",
       "            -7.9018e+00,  1.5332e+01],\n",
       "           ...,\n",
       "           [ 2.1701e+01, -4.9699e+00, -1.5196e+01,  ..., -1.2608e+01,\n",
       "            -2.9665e+01, -2.8344e+00],\n",
       "           [-2.7764e+01, -1.1558e+01,  2.1414e+01,  ..., -2.0382e+01,\n",
       "             1.4446e+01,  2.8969e+01],\n",
       "           [-2.6366e+01, -2.2574e+01, -2.0341e+00,  ..., -2.6535e+01,\n",
       "             5.0701e+00, -1.9766e+01]],\n",
       "\n",
       "          [[-1.9075e+01, -8.0272e+00, -2.7283e+00,  ..., -1.4842e+01,\n",
       "            -1.5040e+01,  3.6856e+00],\n",
       "           [-2.2716e+01, -7.9660e+00,  4.9679e+00,  ..., -3.7975e+00,\n",
       "             2.4438e+00, -1.7929e+01],\n",
       "           [-2.3308e+01, -1.1190e+01, -8.9888e+00,  ...,  2.4150e+01,\n",
       "             1.7011e+01,  1.1784e+01],\n",
       "           ...,\n",
       "           [ 5.6591e-01, -6.3842e+00,  2.4017e+01,  ...,  2.7552e+01,\n",
       "             2.5411e+01,  3.7327e+00],\n",
       "           [ 1.7179e+00, -2.4470e+01,  7.5922e-01,  ...,  2.6079e-01,\n",
       "             9.8388e+00, -3.1953e+00],\n",
       "           [-8.0299e+00, -2.1238e+01,  1.6375e+01,  ...,  4.5360e+01,\n",
       "             2.8393e+00, -3.4362e+01]],\n",
       "\n",
       "          [[ 1.6208e+01, -1.3837e+01,  8.5144e-01,  ..., -1.0817e+01,\n",
       "             3.4897e+01,  6.8206e+00],\n",
       "           [-3.1354e-01, -1.2947e+01, -1.5863e+01,  ..., -1.1471e+01,\n",
       "            -1.4326e+00,  1.3964e+01],\n",
       "           [-7.6486e+00, -2.6234e+01,  2.9283e+01,  ..., -2.6771e+00,\n",
       "             6.6459e+00,  2.1746e+01],\n",
       "           ...,\n",
       "           [-1.2062e+01,  3.4836e+01, -1.5974e+01,  ...,  7.2707e+00,\n",
       "            -1.6994e+01, -9.6398e+00],\n",
       "           [-2.3717e+01,  2.7575e+01, -2.9176e+01,  ...,  2.6785e+01,\n",
       "             3.4637e+01,  1.4359e+01],\n",
       "           [ 5.6186e+00,  4.6176e+01,  1.0734e+01,  ...,  2.0002e+01,\n",
       "             3.0891e+01, -3.9020e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 2.8599e+01, -4.6867e+01, -2.7098e+01,  ..., -3.3584e+01,\n",
       "             2.3556e+01,  1.7929e+00],\n",
       "           [ 2.2372e-02, -3.7277e+01, -1.3122e+01,  ..., -4.3018e+01,\n",
       "            -1.1520e+01, -1.7898e+00],\n",
       "           [-1.9632e+01, -2.9785e+01, -1.3223e+01,  ..., -4.9544e+01,\n",
       "            -8.1881e+00, -5.7976e-01],\n",
       "           ...,\n",
       "           [ 1.5504e+01,  2.7766e+01,  1.6336e+01,  ...,  2.2665e+00,\n",
       "             1.1048e+01,  6.9441e+00],\n",
       "           [ 2.3413e+00, -6.6955e+00, -1.3161e+01,  ...,  4.8350e+00,\n",
       "            -3.6373e+01,  2.5109e+00],\n",
       "           [ 5.9422e+00, -3.5420e+01, -9.8712e+00,  ...,  8.1874e+00,\n",
       "             1.3766e+01,  1.9304e+01]],\n",
       "\n",
       "          [[ 2.5668e+01, -1.5905e+01, -3.2123e+01,  ...,  2.8969e+01,\n",
       "             1.7627e+01,  1.9342e+01],\n",
       "           [ 6.8247e+00, -4.4013e+00, -5.5543e+01,  ...,  4.2838e+01,\n",
       "            -2.2596e+01,  1.0639e+01],\n",
       "           [-5.1776e+00, -1.5780e+00,  2.0286e+00,  ...,  9.5574e+00,\n",
       "            -2.4046e+01, -9.6039e+00],\n",
       "           ...,\n",
       "           [-6.1578e+00, -2.5167e+01, -1.3814e+01,  ..., -1.5346e-01,\n",
       "             4.2753e+01,  1.2074e+01],\n",
       "           [-5.5772e-01,  1.0696e+01, -1.3190e+01,  ..., -3.5321e+01,\n",
       "            -2.1748e+01,  2.4029e+01],\n",
       "           [ 8.0264e+00, -2.2883e+01, -3.0073e+01,  ...,  1.7513e+01,\n",
       "            -2.6798e+01, -8.5506e+00]],\n",
       "\n",
       "          [[-1.7903e+01, -8.4557e-01, -1.9705e+01,  ...,  3.6344e+00,\n",
       "             2.7150e+01, -1.3614e+01],\n",
       "           [ 6.0199e+00, -1.0143e+01,  4.8077e+01,  ...,  8.6949e+00,\n",
       "            -6.1016e+00, -2.4767e+01],\n",
       "           [-3.8141e+00,  2.5840e+01, -1.6484e+01,  ..., -9.4197e+00,\n",
       "            -8.7922e+00, -1.4564e+01],\n",
       "           ...,\n",
       "           [-2.6870e+01, -1.7012e+01, -4.7075e+00,  ...,  1.0512e+01,\n",
       "            -4.4772e+00,  1.1609e+01],\n",
       "           [-3.0689e+01, -1.9170e+00,  3.1441e+01,  ..., -4.1835e+01,\n",
       "            -7.1403e+00, -2.1462e+01],\n",
       "           [-1.2852e+01,  1.4431e+01,  1.7222e+01,  ...,  1.0117e+01,\n",
       "            -7.3001e+00,  5.0193e-01]]],\n",
       "\n",
       "\n",
       "         [[[ 2.1596e+01, -3.2855e+01,  1.0585e+00,  ..., -1.4564e+01,\n",
       "            -2.6468e+01, -2.9532e+01],\n",
       "           [ 2.1065e+01, -2.0668e+01, -6.6598e+00,  ..., -1.8511e+00,\n",
       "             2.4140e+01, -6.3897e+00],\n",
       "           [ 1.0819e+01, -1.4846e-01,  1.2418e+01,  ..., -2.0599e+01,\n",
       "             1.6689e+00, -1.4937e+01],\n",
       "           ...,\n",
       "           [-1.6521e+01,  3.1992e+01, -6.1289e+00,  ..., -2.8736e+01,\n",
       "            -1.2914e+01, -2.5697e+01],\n",
       "           [-3.8329e+01,  1.0580e+01, -1.6258e+00,  ..., -2.0867e+01,\n",
       "            -3.2745e+01,  3.4795e+01],\n",
       "           [ 2.4828e+01,  2.1763e+01, -2.0105e+01,  ...,  1.2022e+01,\n",
       "             1.5122e-01,  1.2923e+01]],\n",
       "\n",
       "          [[-2.1966e+00,  1.3431e+01, -7.4290e+00,  ..., -8.6322e+00,\n",
       "             1.1168e+01,  2.0066e+01],\n",
       "           [ 8.7319e+00,  3.7292e+00,  3.5424e+00,  ...,  1.7815e+01,\n",
       "             1.2495e+01,  5.0757e+01],\n",
       "           [-2.4398e+01,  1.8115e+01, -1.8916e+01,  ...,  2.8629e+00,\n",
       "            -1.7975e+01, -1.5979e+01],\n",
       "           ...,\n",
       "           [ 2.0729e+01, -2.1113e+01,  2.2688e+01,  ...,  1.2700e+01,\n",
       "            -1.8552e+00, -3.3973e+00],\n",
       "           [-6.6868e+00, -1.0851e+01, -1.4673e+01,  ...,  7.4659e+00,\n",
       "             1.4539e+01,  2.8540e+01],\n",
       "           [-8.3602e+00, -5.1140e+01, -3.6011e+00,  ...,  2.8740e+01,\n",
       "             1.1896e+01, -3.7937e+01]],\n",
       "\n",
       "          [[ 2.2183e+01,  3.7827e+01,  3.0893e+01,  ...,  2.0197e+01,\n",
       "            -2.7907e+00, -8.5597e+00],\n",
       "           [ 5.0033e+00,  2.3427e+01,  3.8354e+00,  ..., -3.7609e+01,\n",
       "            -1.9290e+01, -7.0733e-01],\n",
       "           [ 1.2804e+01, -3.5860e+01,  2.0191e+00,  ..., -5.2221e+00,\n",
       "            -1.5709e+01, -3.5996e+00],\n",
       "           ...,\n",
       "           [-2.3731e+00,  1.3158e+01,  5.6846e+01,  ...,  1.9265e+01,\n",
       "             2.1161e+00, -8.3443e+00],\n",
       "           [ 6.7029e+00,  4.1512e+01, -1.0523e+00,  ..., -3.1464e+01,\n",
       "             2.5205e+01,  2.4915e+01],\n",
       "           [-7.4494e+00, -1.5337e+01, -1.5977e+00,  ...,  2.0077e+01,\n",
       "            -2.7139e+01, -3.4443e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.5220e+01, -3.3831e+01,  1.2497e+01,  ...,  1.7618e+00,\n",
       "             1.4333e+01, -1.2098e+01],\n",
       "           [ 1.6951e+01, -1.0830e+01,  2.1176e+01,  ...,  2.3962e+01,\n",
       "            -3.1418e+01,  2.5855e+01],\n",
       "           [-1.6774e+00,  1.1656e+01, -2.7469e+01,  ..., -2.1420e+01,\n",
       "            -3.6560e+00, -3.6514e+00],\n",
       "           ...,\n",
       "           [-1.1274e+01, -1.9327e+01,  3.8072e+01,  ...,  1.1130e+01,\n",
       "            -2.0518e+01, -6.6437e+00],\n",
       "           [-4.9217e+00,  2.6420e+01, -1.2419e+01,  ..., -1.1824e+01,\n",
       "            -8.1176e+00, -3.1557e+01],\n",
       "           [ 8.2471e+00,  3.3843e+01, -1.6147e+01,  ..., -2.1615e+01,\n",
       "            -4.3929e+00, -3.8231e+00]],\n",
       "\n",
       "          [[ 2.3089e+01, -2.0157e+01,  2.0351e+01,  ...,  9.8304e+00,\n",
       "             1.7136e+01, -1.2840e+01],\n",
       "           [-2.2120e+01, -1.0108e+00, -1.7014e+01,  ..., -1.2635e+01,\n",
       "             2.1888e+00, -3.6061e+01],\n",
       "           [ 1.0972e+01, -1.8626e+01, -3.6692e+00,  ..., -2.0680e+01,\n",
       "            -1.2872e+01,  1.1103e+01],\n",
       "           ...,\n",
       "           [-4.2914e+01, -3.3500e+01,  2.1014e+00,  ..., -2.6130e+01,\n",
       "            -1.8057e+00,  4.5757e+01],\n",
       "           [ 1.7259e+00,  1.1409e+01,  4.4012e+01,  ..., -6.9852e+00,\n",
       "            -1.9423e+00, -1.3528e+01],\n",
       "           [-1.1624e+01, -1.1397e+01, -3.7499e+01,  ...,  1.5624e-01,\n",
       "             3.4665e+00,  4.6802e+00]],\n",
       "\n",
       "          [[ 2.4963e+01, -1.4048e+01, -1.3675e+00,  ..., -1.1787e+00,\n",
       "            -6.8066e+00,  2.9660e+01],\n",
       "           [ 8.1655e+00, -3.6068e+00,  1.0754e+00,  ...,  2.5100e+00,\n",
       "             1.3777e+01, -2.9008e+01],\n",
       "           [ 2.8362e+01,  2.1481e+01,  1.2931e+01,  ..., -2.1906e+01,\n",
       "             5.9766e+00,  2.9777e+00],\n",
       "           ...,\n",
       "           [ 1.5703e+01,  2.3771e+01, -5.8411e+00,  ...,  4.2704e+00,\n",
       "             4.8301e+00, -5.1918e+01],\n",
       "           [ 9.1160e+00, -1.1753e+01,  1.7914e+00,  ..., -1.4600e+01,\n",
       "            -1.3697e+01,  1.4241e+01],\n",
       "           [-1.1773e+01,  1.1651e+01, -3.6651e+00,  ...,  1.3886e+01,\n",
       "             1.3920e+01, -3.0547e+01]]],\n",
       "\n",
       "\n",
       "         [[[-5.8023e+00,  7.3468e+00, -1.3278e+01,  ...,  7.3537e+00,\n",
       "            -1.5450e+01, -9.6860e+00],\n",
       "           [-1.1515e+01, -3.1208e+00, -3.0591e+00,  ..., -1.8347e+01,\n",
       "            -3.8034e+01,  2.1083e+01],\n",
       "           [-1.9675e+01,  2.7420e+00,  1.8449e+01,  ..., -2.0700e+01,\n",
       "             2.3693e+00,  2.0830e+01],\n",
       "           ...,\n",
       "           [-3.4071e+00, -1.2889e+01,  4.0382e+01,  ...,  3.1534e+01,\n",
       "             1.6448e+01,  2.1326e+01],\n",
       "           [ 2.7452e+01, -8.6994e+00, -1.8018e+01,  ...,  2.4052e+01,\n",
       "             3.9581e+01,  2.4980e+01],\n",
       "           [-9.5949e+00,  4.9704e+00, -8.5575e+00,  ...,  1.2338e+00,\n",
       "            -2.6328e+01, -5.4489e+00]],\n",
       "\n",
       "          [[ 7.0364e+00,  1.7035e+01, -1.5720e+01,  ..., -1.0260e+00,\n",
       "            -3.9254e+00,  5.9598e+00],\n",
       "           [ 3.3485e-01,  1.7691e+00,  9.5977e+00,  ...,  4.4329e+00,\n",
       "             3.1103e+01,  2.7453e+00],\n",
       "           [ 2.3651e+01, -4.0139e+01,  3.6830e+01,  ..., -4.5885e+01,\n",
       "             3.1717e+01,  2.3522e-01],\n",
       "           ...,\n",
       "           [ 2.9153e+01, -4.6671e-01, -3.4322e+01,  ...,  1.6344e+01,\n",
       "            -1.4092e+01, -2.9438e+01],\n",
       "           [ 7.6201e+00, -2.9194e+00,  2.3669e+01,  ..., -2.4576e+01,\n",
       "             4.0620e+00, -1.7390e+01],\n",
       "           [ 7.4590e+00,  1.2574e+01, -3.4488e+01,  ...,  2.2126e+01,\n",
       "             4.6169e+01, -2.6437e+01]],\n",
       "\n",
       "          [[ 1.8478e+01,  2.8791e+01,  1.7645e+00,  ...,  3.6787e+01,\n",
       "             6.3776e+00,  7.8225e+00],\n",
       "           [-1.9083e+01,  2.3541e+01, -2.6790e+01,  ..., -1.8134e+01,\n",
       "            -6.6634e+00,  3.8249e+01],\n",
       "           [-1.3173e+01, -2.1847e+01, -1.9194e+01,  ..., -2.0890e+01,\n",
       "             4.4939e+00, -1.9879e+01],\n",
       "           ...,\n",
       "           [ 1.5270e+01, -2.0682e+01, -1.9591e+01,  ..., -2.7214e+01,\n",
       "            -1.2088e-01,  3.0713e+01],\n",
       "           [-1.6468e+01, -2.8186e-01, -3.4176e+01,  ...,  1.6278e+01,\n",
       "            -2.2600e+01,  1.7734e+01],\n",
       "           [-2.1183e+01, -2.8731e+01, -5.4226e+00,  ...,  1.0157e+01,\n",
       "             1.0799e+01,  2.8766e+01]],\n",
       "\n",
       "          ...,\n",
       "\n",
       "          [[ 1.2276e+01, -7.5187e+00, -5.4948e+01,  ...,  4.7049e+01,\n",
       "            -1.5492e+01, -2.5511e+00],\n",
       "           [-3.3518e+01, -7.5954e+00,  2.9293e+01,  ...,  5.3828e+00,\n",
       "            -2.2245e+01,  7.0798e+00],\n",
       "           [ 1.2267e+01,  2.8124e+01,  1.5601e+00,  ..., -7.1475e+00,\n",
       "             2.6865e+01, -6.8791e+00],\n",
       "           ...,\n",
       "           [ 1.2467e+01,  2.3620e+01, -4.5572e-01,  ...,  1.7100e+01,\n",
       "             3.9053e+01,  2.1912e+01],\n",
       "           [ 2.0617e+01, -2.1510e+00,  1.9045e+01,  ...,  2.8664e+01,\n",
       "             1.3189e+01, -6.0600e+01],\n",
       "           [-1.1120e+01,  1.9631e+01,  3.3826e+01,  ..., -2.9132e+01,\n",
       "             2.2699e+01, -2.2234e+00]],\n",
       "\n",
       "          [[-9.0135e-01, -2.4605e+01,  1.8034e+01,  ..., -1.1859e+01,\n",
       "             4.1980e+00, -1.6259e+01],\n",
       "           [-3.1686e+01, -1.1413e+01, -2.8415e+01,  ..., -1.2678e+01,\n",
       "             3.3869e+01, -1.9314e+01],\n",
       "           [ 2.9215e+01,  7.6377e+00, -1.9469e+01,  ..., -8.3158e+00,\n",
       "             6.7289e+00, -2.2941e+01],\n",
       "           ...,\n",
       "           [ 1.8493e+01, -1.6573e+01, -3.9948e+01,  ..., -2.6312e+01,\n",
       "             3.7317e+01, -1.8143e+01],\n",
       "           [-3.2424e+01,  2.5779e+01,  3.0072e-01,  ..., -1.3128e+01,\n",
       "             1.9887e+01, -2.6398e+01],\n",
       "           [ 3.8948e+01,  2.2795e+01,  4.9344e+00,  ...,  6.5577e+00,\n",
       "            -8.3507e+00,  2.6005e+01]],\n",
       "\n",
       "          [[ 1.3368e+01,  1.5424e+01,  1.5301e-01,  ...,  4.0589e+00,\n",
       "             1.6071e+01, -2.3438e-02],\n",
       "           [ 3.8557e+00, -3.1724e+01, -2.8947e+01,  ..., -5.9902e+00,\n",
       "            -3.6661e+01,  1.3849e+01],\n",
       "           [-6.5759e+01,  9.8195e+00,  5.4618e+01,  ...,  1.4810e+01,\n",
       "            -4.5196e+01,  4.6851e+01],\n",
       "           ...,\n",
       "           [ 3.0696e+00,  6.3599e+00,  4.1662e+01,  ..., -6.5418e+00,\n",
       "             1.0631e+01, -7.7052e+00],\n",
       "           [-5.6639e+00, -3.3200e+01, -3.9744e+00,  ...,  2.6833e+01,\n",
       "             3.8776e+00, -4.5282e+00],\n",
       "           [-1.4714e+01,  9.2076e+00,  3.9911e+01,  ..., -3.3049e+00,\n",
       "            -2.0085e+00, -9.4762e+00]]]]])"
      ]
     },
     "execution_count": 4,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "filters = autograd.Variable(torch.randn(33, 16, 3, 3, 3))\n",
    "inputs = autograd.Variable(torch.randn(20, 16, 50, 10, 20))\n",
    "F.conv3d(inputs, filters)"
   ]
  },
  {
   "cell_type": "markdown",
   "metadata": {},
   "source": [
    "### torch.nn.functional.conv_transpose1d(input, weight, bias=None, stride=1, padding=0, output_padding=0)\n",
    "### torch.nn.functional.conv_transpose2d(input, weight, bias=None, stride=1, padding=0, output_padding=0)\n",
    "\n",
    "在由几个输入平面组成的输入图像上应用二维转置卷积，有时也称为“去卷积”。 有关详细信息和输出形状，请参阅 ConvTranspose2d 。\n",
    "\n",
    "参数： \n",
    "- input – 输入张量的形状 (minibatch x in_channels x iH x iW) \n",
    "- weight – 过滤器的形状 (in_channels x out_channels x kH x kW) \n",
    "- bias – 可选偏置的形状 (out_channels) \n",
    "- stride – 卷积 核的步长，可以是单个数字或一个元组 (sh x sw)。默认: 1 \n",
    "- padding – 输入上隐含零填充。可 以是单个数字或元组。 (padh x padw)。默认: 0 \n",
    "- groups – 将输入分成组， in_channels 应该被 组数除尽 \n",
    "- output_padding – 0 <= padding <stride的零填充，应该添加到输出。可以是单个数 字或元组。默认值：0"
   ]
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.7.2"
  },
  "toc": {
   "base_numbering": 1,
   "nav_menu": {},
   "number_sections": true,
   "sideBar": true,
   "skip_h1_title": false,
   "title_cell": "Table of Contents",
   "title_sidebar": "Contents",
   "toc_cell": false,
   "toc_position": {},
   "toc_section_display": true,
   "toc_window_display": false
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
